diff --git a/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts b/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts
index ad446753f3a..c083c16c8d3 100644
--- a/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts
+++ b/src/agents/pi-tools.before-tool-call.integration.e2e.test.ts
@@ -503,7 +503,11 @@ describe("before_tool_call hook integration for client tools", () => {
           namespace: "policy",
           value: { gate: "client" },
         }),
-      ).resolves.toMatchObject({ ok: true });
+      ).resolves.toEqual({
+        ok: true,
+        key: "agent:main:client",
+        value: { gate: "client" },
+      });
 
       const [tool] = toClientToolDefinitions(
         [