diff --git a/.github/workflows/rust_tests.yml b/.github/workflows/rust_tests.yml
index 9ad3c648..ed031167 100644
--- a/.github/workflows/rust_tests.yml
+++ b/.github/workflows/rust_tests.yml
@@ -25,8 +25,7 @@ jobs:
 
       - name: Build wasm module
         run: |
-         cd prompt_gateway && cargo build --release --target=wasm32-wasip1
-         cd llm_gateway && cargo build --release --target=wasm32-wasip1
+         cargo build --release --target=wasm32-wasip1 -p llm_gateway -p prompt_gateway
 
       - name: Run unit tests
         run: cargo test --lib
diff --git a/demos/use_cases/preference_based_routing/arch_config.yaml b/demos/use_cases/preference_based_routing/arch_config.yaml
index d31c56dd..f4366201 100644
--- a/demos/use_cases/preference_based_routing/arch_config.yaml
+++ b/demos/use_cases/preference_based_routing/arch_config.yaml
@@ -2,6 +2,8 @@ version: "0.1-beta"
 
 routing:
   model: gpt-4o
+  # model: archgw-router
+  # model: claude-3.7
 
 listeners:
   egress_traffic: