diff --git a/agenta-cli/tests/run_tests.sh b/agenta-cli/tests/run_tests.sh
index 57ccf07b3..66494be5b 100755
--- a/agenta-cli/tests/run_tests.sh
+++ b/agenta-cli/tests/run_tests.sh
@@ -2,5 +2,5 @@
 
 set -e
 
-OPENAI_API_KEY=sk-xxxxxxxx AGENTA_HOST=http://localhost AGENTA_API_KEY=xxxxx.xxxxxxxxxxxxx pytest management/ -v -m variant_management # run variant_management tests
-
+OPENAI_API_KEY=sk-xxxxx AGENTA_HOST=http://localhost AGENTA_API_KEY=xxxx.xxxxxxxxxxxxxxx pytest -n 2 -v ./management/*
+BASE_URL=http://127.0.0.1 AGENTA_HOST=http://localhost AGENTA_API_KEY=xxxx.xxxxxxxxxxxxxxx pytest -v ./sdk_routing/*