diff --git a/nix/default.nix b/nix/default.nix
index 0111c60..3a62ef3 100644
--- a/nix/default.nix
+++ b/nix/default.nix
@@ -40,7 +40,12 @@ let
   devShells.default = mkShell {
     name = "mu-shell";
 
-    packages = (attrValues packages) ++ [
+    packages = [
+      packages.cargo-risczero
+      packages.r0vm
+      packages.risc0-rust
+      packages.rustup-mock
+
       final.cargo-nextest
       final.cargo-watch
     ];
diff --git a/nix/packages/mugraph-node.nix b/nix/packages/mugraph-node.nix
index dbf9ca4..d1d65e8 100644
--- a/nix/packages/mugraph-node.nix
+++ b/nix/packages/mugraph-node.nix
@@ -11,7 +11,9 @@ mugraph.risc0Platform.buildRustPackage {
   version = "0.0.1";
   src = ../..;
 
-  env.RISC0_RUST_SRC = "${mugraph.rust}/lib/rustlib/src/rust";
+  prePatch = ''
+    export RISC0_RUST_SRC="${mugraph.rust}/lib/rustlib/src/rust"
+  '';
 
   cargoLock.lockFile = ../../Cargo.lock;
   nativeBuildInputs = [ makeWrapper ];