diff --git a/jupyter/flake.nix b/jupyter/flake.nix
index cb8b521..5b473b3 100644
--- a/jupyter/flake.nix
+++ b/jupyter/flake.nix
@@ -21,6 +21,7 @@
           [python311 virtualenv]
           ++ (with pkgs.python311Packages; [
             pip
+            python-lsp-server
             venvShellHook
             requests
             jupyter
diff --git a/python/flake.nix b/python/flake.nix
index dac0c7f..b8682df 100644
--- a/python/flake.nix
+++ b/python/flake.nix
@@ -16,12 +16,11 @@
   in {
     devShells = forEachSupportedSystem ({pkgs}: {
       default = pkgs.mkShell {
-        venvDir = "venv";
         packages = with pkgs;
           [python311]
           ++ (with pkgs.python311Packages; [
             pip
-            venvShellHook
+            python-lsp-server
           ]);
       };
     });