diff --git a/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py b/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py
index a3702bbd29b..4ae815186c3 100644
--- a/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py
+++ b/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py
@@ -481,6 +481,5 @@ class TestShortCircuitEntryPoint:
 
         assert result is not None
         mock_search.assert_called_once_with(expected_query)
-        tool_use_block = next((b for b in result["content"] if b["type"] == "server_tool_use"), None)
-        if tool_use_block is not None:
-            assert tool_use_block["input"]["query"] == expected_query
+        tool_use_block = next(b for b in result["content"] if b["type"] == "server_tool_use")
+        assert tool_use_block["input"]["query"] == expected_query