diff --git a/flake.nix b/flake.nix
index 735ba49b..2ddfedf3 100644
--- a/flake.nix
+++ b/flake.nix
@@ -62,7 +62,7 @@
 
       nixosConfigurations = import ./nixos-configs.nix { inherit lib; };
       overlays.default = overlays.overlay;
-      devShells.${system} = import ./shells.nix { inherit inputs lib pkgs; };
+      devShells.${system} = import ./shells.nix { inherit inputs lib system pkgs; };
       packages.${system} = privateRepo;
       templates = import ./templates.nix { inherit lib; };
     };
diff --git a/shells.nix b/shells.nix
index bd0c1c26..0923b79a 100644
--- a/shells.nix
+++ b/shells.nix
@@ -1,5 +1,5 @@
-{ inputs, lib, pkgs, ... }: with lib;
-with lib.my; let
+{ inputs, lib, pkgs, system, ... }:
+let
   shellPkgs = import inputs.nixpkgs-unstable {
     inherit system;