diff --git a/tests/test_context_aggregators_universal.py b/tests/test_context_aggregators_universal.py
index 59aab4745..e94a68ad1 100644
--- a/tests/test_context_aggregators_universal.py
+++ b/tests/test_context_aggregators_universal.py
@@ -806,7 +806,7 @@ class TestLLMAssistantAggregator(unittest.IsolatedAsyncioTestCase):
                 function_name="get_weather",
                 tool_call_id="1",
                 arguments={"location": "Los Angeles"},
-                cancel_on_interruption=False,
+                cancel_on_interruption=True,
             ),
             SleepFrame(),
             FunctionCallResultFrame(
@@ -838,7 +838,7 @@ class TestLLMAssistantAggregator(unittest.IsolatedAsyncioTestCase):
                 function_name="get_weather",
                 tool_call_id="1",
                 arguments={"location": "Los Angeles"},
-                cancel_on_interruption=False,
+                cancel_on_interruption=True,
             ),
             SleepFrame(),
             FunctionCallResultFrame(