diff --git a/tests/integration/agents/test_agents.py b/tests/integration/agents/test_agents.py
index 23529f91e..07ba7bb01 100644
--- a/tests/integration/agents/test_agents.py
+++ b/tests/integration/agents/test_agents.py
@@ -68,6 +68,7 @@ def agent_config(llama_stack_client, text_model_id):
                 "temperature": 0.0001,
                 "top_p": 0.9,
             },
+            "max_tokens": 512,
         },
         tools=[],
         input_shields=available_shields,
@@ -88,6 +89,7 @@ def agent_config_without_safety(text_model_id):
                 "temperature": 0.0001,
                 "top_p": 0.9,
             },
+            "max_tokens": 512,
         },
         tools=[],
         enable_session_persistence=False,
@@ -198,6 +200,7 @@ def test_tool_config(agent_config):
                 "temperature": 1.0,
                 "top_p": 0.9,
             },
+            "max_tokens": 512,
         },
         toolgroups=[],
         enable_session_persistence=False,