diff --git a/nilla.nix b/nilla.nix
index 7c7f4e0..a1841f5 100644
--- a/nilla.nix
+++ b/nilla.nix
@@ -69,6 +69,9 @@ nilla.create (
             ...
           }:
           mkShell {
+            shellHook = ''
+              export ZED_PREDICT_EDITS_URL=http://localhost:9000/predict_edits  
+            '';
             packages = [
               pkgs.python314
               pkgs.black