diff --git a/tools/bazel b/tools/bazel
index 78e4624..0f235c3 100755
--- a/tools/bazel
+++ b/tools/bazel
@@ -34,9 +34,8 @@
   # Jump to project root since bwrap hangs if we aren't there
   cd "${DIR}/../"
 
-  export COMMAND="bazel $*"
   export PWD="$OLDPWD"
-  exec nix-shell
+  exec nix-shell --command "bazel $*"
 fi
 
 prechecks() {