diff --git a/flake.nix b/flake.nix
index 019f27c..0bd23c6 100644
--- a/flake.nix
+++ b/flake.nix
@@ -10,9 +10,7 @@
   {
     devShells."${system}" =
     let
-      pkgs = import nixpkgs {
-        inherit system;
-      };
+      pkgs = nixpkgs.legacyPackages."${system}";
     in
       pkgs.mkShellNoCC {
         packages = with pkgs; [
@@ -31,10 +29,10 @@
             ];
           })
         ];
+        shellHook = ''
+          echo "starting codium"
+          codium .
+        '';
       };
-      shellHook = ''
-        echo "starting codium"
-        codium .
-      '';
   };
 }