diff --git a/agents-api/agents_api/activities/task_steps/tool_call_step.py b/agents-api/agents_api/activities/task_steps/tool_call_step.py
index a93e5bb66..a576bbbeb 100644
--- a/agents-api/agents_api/activities/task_steps/tool_call_step.py
+++ b/agents-api/agents_api/activities/task_steps/tool_call_step.py
@@ -4,9 +4,8 @@
 from beartype import beartype
 from temporalio import activity
 
-from agents_api.activities.task_steps import base_evaluate
-from agents_api.autogen.Tasks import ToolCallStep
-
+from ...activities.task_steps import base_evaluate
+from ...autogen.openapi_model import ToolCallStep
 from ...common.protocol.tasks import (
     StepContext,
     StepOutcome,