diff --git a/default.nix b/default.nix
index 6e487c9..d3c9b62 100644
--- a/default.nix
+++ b/default.nix
@@ -16,7 +16,7 @@ buildDunePackage {
   src = ./.;
   duneVersion = "3";
   postPatch = ''
-    substituteInPlace direnvrc --replace "flake_env" "$out/bin/flake_env"
+    substituteInPlace direnvrc --replace "@flake_env@" "$out/bin/flake_env"
   '';
   postInstall = ''
     install -m400 -D direnvrc $out/share/flake_env/direnvrc
diff --git a/direnvrc b/direnvrc
index a6ffd2c..0d72aaf 100644
--- a/direnvrc
+++ b/direnvrc
@@ -39,7 +39,7 @@ use_flake_env() {
   local old_xdg_data_dirs=${XDG_DATA_DIRS:-}
 
   local ld=$(direnv_layout_dir)
-  eval $(flake_env "$1" "$ld")
+  eval $(@flake_env@ "$1" "$ld")
 
   # `nix print-dev-env` will create a temporary directory and use it as TMPDIR
   # We cannot rely on this directory being available at all times,