1 files changed, 9 insertions, 2 deletions
diff --git a/tools.nix b/tools.nix
index 5f90e27..2c942ff 100644
--- a/tools.nix
+++ b/tools.nix
@@ -1,9 +1,13 @@
-{ stdenvNoCC }:
+{ stdenvNoCC, makeWrapper, flakeSelf }:
 stdenvNoCC.mkDerivation {
   name = "kernel-dev-tools";
 
   src = ./tools;
-  
+
+  nativeBuildInputs = [
+    makeWrapper
+  ];
+
   dontConfigure = true;
   dontBuild = true;
 
@@ -16,6 +20,9 @@ stdenvNoCC.mkDerivation {
       install -m0555 "$script" $out/bin/
     done
 
+    wrapProgram $out/bin/enter-kernel-dev \
+      --set FLAKE_DIR "${flakeSelf}"
+
     runHook preInstall
   '';
 }