diff --git a/Dockerfile b/Dockerfile
index db68400..f922893 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -72,6 +72,7 @@ RUN wget -O /opt/ijava-kernel.zip https://github.com/SpencerPark/IJava/releases/
 RUN curl -fsSL https://code-server.dev/install.sh | sh
 RUN code-server --install-extension redhat.vscode-yaml \
         --install-extension ms-python.python \
+        --install-extension bungcip.better-toml \
         --install-extension vscjava.vscode-java-pack \
         --install-extension ginfuru.ginfuru-better-solarized-dark-theme \
         --install-extension oderwat.indent-rainbow \
diff --git a/gpu.dockerfile b/gpu.dockerfile
index 23106b1..a797032 100644
--- a/gpu.dockerfile
+++ b/gpu.dockerfile
@@ -43,6 +43,7 @@ RUN pip3 install --upgrade pip && \
 RUN curl -fsSL https://code-server.dev/install.sh | sh
 RUN code-server --install-extension redhat.vscode-yaml \
         --install-extension ms-python.python \
+        --install-extension bungcip.better-toml \
         --install-extension ginfuru.ginfuru-better-solarized-dark-theme \
         --install-extension oderwat.indent-rainbow \
         --install-extension mechatroner.rainbow-csv \