diff --git a/tests/unittests/flows/llm_flows/test_functions_request_euc.py b/tests/unittests/flows/llm_flows/test_functions_request_euc.py
index 6f8f112..afb3b73 100644
--- a/tests/unittests/flows/llm_flows/test_functions_request_euc.py
+++ b/tests/unittests/flows/llm_flows/test_functions_request_euc.py
@@ -360,6 +360,9 @@ def test_function_get_auth_response_partial():
       [
           types.Part.from_text(text='response2'),
       ],
+      [
+          types.Part.from_text(text='final response'),
+      ],
   ]
 
   mock_model = testing_utils.MockModel.create(responses=responses)