diff --git a/agent.py b/agent.py
index 5deddfb..15a45d3 100644
--- a/agent.py
+++ b/agent.py
@@ -556,7 +556,7 @@ class CustomerIntegrationEngineer(A2AAgent[CustomerIntegrationEngineerConfig, No
             drift_detected=drift,
         )
 
-    @a2a.tool(description="Run dry-run or allowlisted acceptance checks using endpoint metadata and secret references", input_schema=_PUBLIC_REQUEST_SCHEMA, timeout_seconds=180, idempotent=True, max_retries=0, cost_class="deterministic")
+    @a2a.tool(description="Run dry-run or allowlisted acceptance checks using endpoint metadata and secret references", input_schema=_tool_input_schema(RunAcceptanceRequest), timeout_seconds=180, idempotent=True, max_retries=0, cost_class="deterministic")
     async def run_acceptance(self, ctx: RunContext[NoAuth], request: RunAcceptanceRequest) -> Any:
         dry_run = _effective_dry_run(self, request.dry_run)
         live_network_block = _live_network_unavailable_reason(self)