diff --git a/model_server/tests/core/test_function_calling.py b/model_server/tests/core/test_function_calling.py
index 7aba557b..9f1d0750 100644
--- a/model_server/tests/core/test_function_calling.py
+++ b/model_server/tests/core/test_function_calling.py
@@ -101,7 +101,7 @@ def get_greeting_data():
     ],
 )
 async def test_function_calling(get_data_func):
-    req, intent, hallucination, parameter_gathering = get_data_func()
+    req, intent, hallucination = get_data_func()
     handler_name = "Arch-Function"
     use_agent_orchestrator = False
     model_handler: ArchFunctionHandler = handler_map[handler_name]