diff --git a/agent.py b/agent.py
index ead88d8..d387d24 100644
--- a/agent.py
+++ b/agent.py
@@ -522,7 +522,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No
         )
 
     @a2a.tool(description="Generate a minimal workspace-contained connector scaffold with no deployment or repository mutation", input_schema=_GENERATE_INPUT_SCHEMA, timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic", grant_outputs_prefix="outputs", grant_write_prefixes=("outputs/integrations",))
-    async def generate_connector(self, ctx: RunContext[NoAuth], request: GenerateConnectorRequest) -> GenerateConnectorResult:
+    async def generate_connector(self, ctx: RunContext[NoAuth], request: GenerateConnectorRequest) -> Any:
         req = request.requirements.model_copy(
             update={"dry_run": _effective_dry_run(self, request.dry_run, request.requirements.dry_run)}
         )
@@ -608,7 +608,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No
         )
 
     @a2a.tool(description="Run dry-run or allowlisted acceptance checks using endpoint metadata and secret references", input_schema=_ACCEPTANCE_INPUT_SCHEMA, timeout_seconds=180, idempotent=True, max_retries=0, cost_class="deterministic", grant_outputs_prefix="outputs", grant_write_prefixes=("outputs/integrations",))
-    async def run_acceptance(self, ctx: RunContext[NoAuth], request: RunAcceptanceRequest) -> Any:
+    async def run_acceptance(self, ctx: RunContext[NoAuth], request: RunAcceptanceRequest) -> RunAcceptanceResult:
         dry_run = _effective_dry_run(self, request.dry_run)
         live_network_block = _live_network_unavailable_reason(self)
         audit = [_audit("acceptance_started", request.integration_id, {"endpoint_tests": len(request.endpoint_tests), "dry_run": dry_run, "allow_network": request.allow_network})]