diff --git a/agents-api/tests/test_workflow_helpers.py b/agents-api/tests/test_workflow_helpers.py
index 8cf053d7b..f64f67036 100644
--- a/agents-api/tests/test_workflow_helpers.py
+++ b/agents-api/tests/test_workflow_helpers.py
@@ -1,59 +1,21 @@
 import uuid
-from datetime import timedelta
-from unittest.mock import Mock, call, patch
+from unittest.mock import patch
 
-from agents_api.activities import task_steps
-from agents_api.activities.execute_api_call import execute_api_call
-from agents_api.activities.execute_integration import execute_integration
-from agents_api.activities.execute_system import execute_system
-from agents_api.activities.task_steps.base_evaluate import base_evaluate
 from agents_api.autogen.openapi_model import (
     Agent,
-    ApiCallDef,
-    BaseIntegrationDef,
-    CaseThen,
-    EvaluateStep,
-    Execution,
-    ForeachDo,
-    ForeachStep,
-    GetStep,
-    IfElseWorkflowStep,
-    LogStep,
     MapReduceStep,
-    PromptItem,
-    PromptStep,
-    ReturnStep,
-    SetStep,
-    SwitchStep,
-    SystemDef,
     TaskSpecDef,
-    TaskToolDef,
-    ToolCallStep,
-    Transition,
     TransitionTarget,
-    WaitForInputInfo,
-    WaitForInputStep,
     Workflow,
     YieldStep,
 )
 from agents_api.common.protocol.tasks import (
     ExecutionInput,
-    PartialTransition,
     StepContext,
-    StepOutcome,
 )
-from agents_api.common.retry_policies import DEFAULT_RETRY_POLICY
 from agents_api.common.utils.datetime import utcnow
-from agents_api.env import (
-    debug,
-    temporal_heartbeat_timeout,
-    temporal_schedule_to_close_timeout,
-    testing,
-)
-from agents_api.workflows.task_execution import TaskExecutionWorkflow
-from temporalio.exceptions import ApplicationError
-from ward import raises, test
 from agents_api.workflows.task_execution.helpers import execute_map_reduce_step_parallel
+from ward import raises, test
 
 
 @test("execute_map_reduce_step_parallel: subworkflow step not supported")
@@ -61,7 +23,9 @@ async def _():
     async def _resp():
         return "response"
 
-    subworkflow_step = YieldStep(kind_="yield", workflow='subworkflow', arguments={'test': '$ _'})
+    subworkflow_step = YieldStep(
+        kind_="yield", workflow="subworkflow", arguments={"test": "$ _"}
+    )
 
     step = MapReduceStep(
         kind_="map_reduce",