diff --git a/flake.nix b/flake.nix
index b3ba9d7..dc4f9da 100644
--- a/flake.nix
+++ b/flake.nix
@@ -31,7 +31,7 @@
                 extensions.vscode-marketplace.mhutchie.git-graph
                 extensions.vscode-marketplace.ms-azuretools.vscode-docker
                 extensions.vscode-marketplace.ms-python.vscode-pylance
-                extensions.vscode-marketplace.ms-python.python
+                (extensions.forVSCodeVersion "1.81.1").extensions.vscode-marketplace.ms-python.python
                 extensions.vscode-marketplace.ms-toolsai.jupyter
                 extensions.vscode-marketplace.ms-vscode.cpptools
                 extensions.vscode-marketplace.github.github-vscode-theme