diff --git a/scripts/integration-tests.sh b/scripts/integration-tests.sh
index c0bef8302..bebdbd3ab 100755
--- a/scripts/integration-tests.sh
+++ b/scripts/integration-tests.sh
@@ -141,6 +141,10 @@ fi
 
 # Apply setup-specific environment variables (needed for server startup and tests)
 echo "=== Applying Setup Environment Variables ==="
+
+# the server needs this
+export LLAMA_STACK_INFERENCE_MODE=$INFERENCE_MODE
+
 SETUP_ENV=$(PYTHONPATH=$THIS_DIR/.. python "$THIS_DIR/get_setup_env.py" --suite "$TEST_SUITE" --setup "$TEST_SETUP" --format bash)
 echo "Setting up environment variables:"
 echo "$SETUP_ENV"
diff --git a/tests/integration/suites.py b/tests/integration/suites.py
index 4aaa53991..bacd7ef52 100644
--- a/tests/integration/suites.py
+++ b/tests/integration/suites.py
@@ -51,6 +51,7 @@ SETUP_DEFINITIONS: dict[str, Setup] = {
         description="Local Ollama provider with text + safety models",
         env={
             "OLLAMA_URL": "http://0.0.0.0:11434",
+            "SAFETY_MODEL": "ollama/llama-guard3:1b",
         },
         defaults={
             "text_model": "ollama/llama3.2:3b-instruct-fp16",