diff --git a/flake.nix b/flake.nix
index 32050cb..099f210 100644
--- a/flake.nix
+++ b/flake.nix
@@ -41,7 +41,8 @@
       devShells.${system}.default = pkgs.mkShell {
         buildInputs = with pkgs; [ (import home-manager { inherit pkgs; }).home-manager just ];
         shellHook = ''
-          export NIXPKGS_PATH=${pkgs.path};
+          ln -s ${pkgs.path} .direnv/nixpkgs;
+          export NIX_PATH=`pwd`/.direnv;
         '';
       };
     };
diff --git a/justfile b/justfile
index 96cf60e..65fe1ac 100644
--- a/justfile
+++ b/justfile
@@ -31,7 +31,7 @@ _switch:
 switch: (_run "_switch")
 
 _update-vscode-extensions:
-    $NIXPKGS_PATH/pkgs/applications/editors/vscode/extensions/update_installed_exts.sh > \
+    $NIX_PATH/nixpkgs/pkgs/applications/editors/vscode/extensions/update_installed_exts.sh > \
         {{ root_dir }}/programs/vscode/extensions.nix
 
 _update: _update-vscode-extensions && _switch