diff --git a/agent.py b/agent.py
index 02595dd..1c298d8 100644
--- a/agent.py
+++ b/agent.py
@@ -432,7 +432,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No
         ConsumerSetupField.config("ARGOCD_APP_REF", label="Argo CD app reference", description="Optional caller-owned Argo CD app reference.", required=False),
     )
 
-    @a2a.tool(description="Capture requirements, produce architecture and field mapping artifacts", input_schema=_PUBLIC_REQUEST_SCHEMA, timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic")
+    @a2a.tool(description="Capture requirements, produce architecture and field mapping artifacts", input_schema=_tool_input_schema(DesignIntegrationRequest), timeout_seconds=120, idempotent=True, max_retries=0, cost_class="deterministic")
     async def design_integration(self, ctx: RunContext[NoAuth], request: DesignIntegrationRequest) -> Any:
         req = request.requirements.model_copy(
             update={"dry_run": _effective_dry_run(self, request.dry_run, request.requirements.dry_run)}