diff --git a/tests/test_integration_llm_features.py b/tests/test_integration_llm_features.py
index 9d0962f19..9a7d04dfd 100644
--- a/tests/test_integration_llm_features.py
+++ b/tests/test_integration_llm_features.py
@@ -75,7 +75,9 @@ class TestIntegrationLLMFeatures:
         )
 
         with patch.object(lite_agent, '_invoke_loop') as mock_invoke:
-            mock_invoke.return_value = response_with_xml
+            from crewai.agents.agent_action import AgentFinish
+            mock_agent_finish = AgentFinish(output=response_with_xml, text=response_with_xml)
+            mock_invoke.return_value = mock_agent_finish
             
             result = lite_agent.kickoff("Analyze this problem")