diff --git a/agent.py b/agent.py
index 0a64209..2321f55 100644
--- a/agent.py
+++ b/agent.py
@@ -635,7 +635,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No
             network_calls_made=network_calls,
         )
 
-    @a2a.tool(description="Prepare rollout, rollback, monitoring, and operator handoff documentation with approval-gated delivery planning", input_schema=_tool_input_schema(PrepareHandoffRequest), timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic")
+    @a2a.tool(description="Prepare rollout, rollback, monitoring, and operator handoff documentation with approval-gated delivery planning", input_schema=_tool_input_schema(PrepareHandoffRequest), output_schema=_HANDOFF_OUTPUT_SCHEMA, timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic")
     async def prepare_handoff(self, ctx: RunContext[NoAuth], request: PrepareHandoffRequest) -> Any:
         dry_run = _effective_dry_run(self, request.dry_run)
         audit = [_audit("handoff_started", request.integration_id, {"dry_run": dry_run, "delivery_actions": len(request.delivery_actions)})]