diff --git a/nixos/modules/services/development/jupyter/default.nix b/nixos/modules/services/development/jupyter/default.nix
index 03d64863fd414..ac5b5f790d793 100644
--- a/nixos/modules/services/development/jupyter/default.nix
+++ b/nixos/modules/services/development/jupyter/default.nix
@@ -10,9 +10,8 @@ let
 
   package = pkgs.python3.withPackages (
     ps:
-    with ps;
     [
-      jupyter
+      cfg.package
     ]
     ++ cfg.extraPackages
   );
@@ -46,6 +45,12 @@ in
       '';
     };
 
+    package = lib.mkPackageOption pkgs [
+      "python3"
+      "pkgs"
+      "jupyter"
+    ] { };
+
     extraPackages = lib.mkOption {
       type = lib.types.listOf lib.types.package;
       default = [ ];