diff --git a/agent.py b/agent.py
index 3b3e865..f0d02e6 100644
--- a/agent.py
+++ b/agent.py
@@ -532,8 +532,8 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No
             blocked_actions=blocked_actions,
         )
 
-    @a2a.tool(description="Validate OpenAPI, JSON Schema, and webhook contracts with deterministic fixtures", input_schema=_tool_input_schema(ValidateContractRequest), output_schema=_VALIDATE_OUTPUT_SCHEMA, timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic")
-    async def validate_contract(self, ctx: RunContext[NoAuth], request: ValidateContractRequest) -> Any:
+    @a2a.tool(description="Validate OpenAPI, JSON Schema, and webhook contracts with deterministic fixtures", input_schema=_tool_input_schema(ValidateContractRequest), timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic")
+    async def validate_contract(self, ctx: RunContext[NoAuth], request: ValidateContractRequest) -> ValidateContractResult:
         dry_run = _effective_dry_run(self, request.dry_run)
         audit = [_audit("contract_validation_started", request.integration_id, {"contracts": len(request.contracts), "dry_run": dry_run})]
         await _emit(ctx, audit[-1])