diff --git a/src/crewai/agent.py b/src/crewai/agent.py
index a64dc489ce..e137a1e412 100644
--- a/src/crewai/agent.py
+++ b/src/crewai/agent.py
@@ -20,8 +20,8 @@
 from crewai.utilities.training_handler import CrewTrainingHandler
 
 
-def mock_agent_ops_provider(*args, **kwargs):
-    def track_agent():
+def mock_agent_ops_provider():
+    def track_agent(*args, **kwargs):
         def noop(f):
             return f
 
diff --git a/src/crewai/utilities/evaluators/task_evaluator.py b/src/crewai/utilities/evaluators/task_evaluator.py
index 1ebcdb5983..ed2df41ce9 100644
--- a/src/crewai/utilities/evaluators/task_evaluator.py
+++ b/src/crewai/utilities/evaluators/task_evaluator.py
@@ -8,8 +8,8 @@
 from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser
 
 
-def mock_agent_ops_provider(*args, **kwargs):
-    def track_agent():
+def mock_agent_ops_provider():
+    def track_agent(*args, **kwargs):
         def noop(f):
             return f