diff --git a/crates/Cargo.toml b/crates/Cargo.toml
index f8bfc0d2..3ba99280 100644
--- a/crates/Cargo.toml
+++ b/crates/Cargo.toml
@@ -1,2 +1,3 @@
 [workspace]
+resolver = "2"
 members = ["llm_gateway", "prompt_gateway", "common"]
diff --git a/crates/llm_gateway/tests/integration.rs b/crates/llm_gateway/tests/integration.rs
index 80ff8d9f..7ec92ccd 100644
--- a/crates/llm_gateway/tests/integration.rs
+++ b/crates/llm_gateway/tests/integration.rs
@@ -7,7 +7,7 @@ use serial_test::serial;
 use std::path::Path;
 
 fn wasm_module() -> String {
-    let wasm_file = Path::new("target/wasm32-wasi/release/llm_gateway.wasm");
+    let wasm_file = Path::new("../target/wasm32-wasi/release/llm_gateway.wasm");
     assert!(
         wasm_file.exists(),
         "Run `cargo build --release --target=wasm32-wasi` first"