diff --git a/tests/e2e/test_ollama_integration.py b/tests/e2e/test_ollama_integration.py
index e193e0bf..6fd8b82e 100644
--- a/tests/e2e/test_ollama_integration.py
+++ b/tests/e2e/test_ollama_integration.py
@@ -115,9 +115,9 @@ async def test_system_prompt_selection():
     assert prompt_with_tools is not None, "Prompt with tools should not be None"
     assert prompt_without_tools is not None, "Prompt without tools should not be None"
 
-    # Both should identify as a local AI assistant
-    assert "local AI assistant" in prompt_with_tools, "Prompt should mention local AI assistant"
-    assert "local AI assistant" in prompt_without_tools, "Prompt should mention local AI assistant"
+    # Both should identify as running locally
+    assert "locally" in prompt_with_tools, "Prompt should mention running locally"
+    assert "locally" in prompt_without_tools, "Prompt should mention running locally"
 
     # Full prompt should mention tools
     assert "tool" in prompt_with_tools.lower(), "Full prompt should mention tools"