diff --git a/modules/languages/python.nix b/modules/languages/python.nix
index ab80ca6..367391e 100644
--- a/modules/languages/python.nix
+++ b/modules/languages/python.nix
@@ -64,7 +64,11 @@ in {
         type = with types; enum (attrNames servers);
         default = defaultServer;
       };
-      package = nvim.types.mkGrammarOption pkgs "python";
+      package = mkOption {
+        description = "python LSP server package";
+        type = types.package;
+        default = servers.${cfg.lsp.server}.package;
+      };
     };
 
     format = {