diff --git a/tests/agent_server/test_goal_loop.py b/tests/agent_server/test_goal_loop.py
index ecc7dd00b7..0f5cdb4a18 100644
--- a/tests/agent_server/test_goal_loop.py
+++ b/tests/agent_server/test_goal_loop.py
@@ -43,7 +43,6 @@ def completion(
         self,
         messages,
         tools=None,
-        _return_metrics=False,
         add_security_risk_prediction=False,
         on_token=None,
         **kwargs,
@@ -53,7 +52,6 @@ def completion(
         return super().completion(
             messages,
             tools,
-            _return_metrics,
             add_security_risk_prediction,
             on_token,
             **kwargs,