diff --git a/shell.nix b/shell.nix
index d7f54c4b..019d92c8 100644
--- a/shell.nix
+++ b/shell.nix
@@ -7,27 +7,24 @@
         ./nix/packages/overlay.nix
       ];
     },
-}: rec {
-  default = pkgs.mkShell {
-    packages = with pkgs; [
-      just
-      jq
-      npins
-      disko
-    ];
-  };
-
-  withXMonad = pkgs.mkShell {
-    inputsFrom = [
-      default
-    ];
-    packages = with pkgs; [
-      (haskellPackages.ghcWithPackages (self: [
-        self.xmonad-contrib
-        self.xmonad-extras
-      ]))
-      haskell-language-server
-      cabal-install
-    ];
-  };
-}
+  withXMonad ? false,
+}: let
+  inherit (pkgs) lib;
+in
+  pkgs.mkShell {
+    packages = with pkgs;
+      [
+        just
+        jq
+        npins
+        disko
+      ]
+      ++ lib.optionals withXMonad [
+        (haskellPackages.ghcWithPackages (self: [
+          self.xmonad-contrib
+          self.xmonad-extras
+        ]))
+        haskell-language-server
+        cabal-install
+      ];
+  }