diff --git a/src/agents/pi-embedded-runner.sanitize-session-history.test-harness.ts b/src/agents/pi-embedded-runner.sanitize-session-history.test-harness.ts
index bb371798420a..1761599cd7ad 100644
--- a/src/agents/pi-embedded-runner.sanitize-session-history.test-harness.ts
+++ b/src/agents/pi-embedded-runner.sanitize-session-history.test-harness.ts
@@ -8,6 +8,7 @@ export type SanitizeSessionHistoryFn = (params: {
   messages: AgentMessage[];
   modelApi: string;
   provider: string;
+  allowedToolNames?: Iterable<string>;
   sessionManager: SessionManager;
   sessionId: string;
   modelId?: string;