diff --git a/agents-api/agents_api/activities/task_steps/raise_complete_async.py b/agents-api/agents_api/activities/task_steps/raise_complete_async.py
index 904b298d..adae15b7 100644
--- a/agents-api/agents_api/activities/task_steps/raise_complete_async.py
+++ b/agents-api/agents_api/activities/task_steps/raise_complete_async.py
@@ -1,13 +1,11 @@
 import base64
+from typing import Any
 
 from beartype import beartype
 from temporalio import activity
 
 from ...autogen.openapi_model import CreateTransitionRequest
-from ...common.protocol.tasks import (
-    StepContext,
-    StepOutcome,
-)
+from ...common.protocol.tasks import StepContext
 from ...common.storage_handler import auto_blob_store
 from .transition_step import original_transition_step
 
@@ -15,7 +13,7 @@
 @activity.defn
 @auto_blob_store
 @beartype
-async def raise_complete_async(context: StepContext, output: StepOutcome) -> None:
+async def raise_complete_async(context: StepContext, output: Any) -> None:
     activity_info = activity.info()
 
     captured_token = base64.b64encode(activity_info.task_token).decode("ascii")