diff --git a/tests/integration/responses/test_tool_responses.py b/tests/integration/responses/test_tool_responses.py
index e7087dcd0..49bcd050b 100644
--- a/tests/integration/responses/test_tool_responses.py
+++ b/tests/integration/responses/test_tool_responses.py
@@ -669,7 +669,7 @@ def test_max_tool_calls_invalid(responses_client, text_model_id):
 
     # Create a response with an invalid max_tool_calls value i.e. 0
     # Handle ValueError from LLS and BadRequestError from OpenAI client
-    with pytest.raises((ValueError, openai.BadRequestError)) as excinfo:
+    with pytest.raises((ValueError, llama_stack_client.BadRequestError, openai.BadRequestError)) as excinfo:
         responses_client.responses.create(
             model=text_model_id,
             input=input,