diff --git a/tests/integration_test_send_message.py b/tests/integration_test_send_message.py
index d91ce8de..d85a4c65 100644
--- a/tests/integration_test_send_message.py
+++ b/tests/integration_test_send_message.py
@@ -464,13 +464,6 @@ def assert_tool_call_response(
         ):
             return
 
-    try:
-        assert len(messages) == expected_message_count, messages
-    except:
-        if "claude-3-7-sonnet" not in llm_config.model:
-            raise
-        assert len(messages) == expected_message_count - 1, messages
-
     # OpenAI gpt-4o-mini can sometimes omit the final AssistantMessage in streaming,
     # yielding the shorter sequence:
     #   Reasoning -> ToolCall -> ToolReturn -> Reasoning -> StopReason -> Usage
@@ -489,6 +482,13 @@ def assert_tool_call_response(
     ):
         return
 
+    try:
+        assert len(messages) == expected_message_count, messages
+    except:
+        if "claude-3-7-sonnet" not in llm_config.model:
+            raise
+        assert len(messages) == expected_message_count - 1, messages
+
     index = 0
     if from_db:
         assert isinstance(messages[index], UserMessage)