diff --git a/src/agents/pi-embedded-runner/tool-split.ts b/src/agents/pi-embedded-runner/tool-split.ts
index 4b82ffdbf6b..50a390ab399 100644
--- a/src/agents/pi-embedded-runner/tool-split.ts
+++ b/src/agents/pi-embedded-runner/tool-split.ts
@@ -1,4 +1,5 @@
 import type { AgentTool } from "@mariozechner/pi-agent-core";
+import type { CreateAgentSessionOptions } from "@mariozechner/pi-coding-agent";
 import { toToolDefinitions } from "../pi-tool-definition-adapter.js";
 
 // We always pass tools via `customTools` so our policy filtering, sandbox integration,
@@ -6,7 +7,7 @@ import { toToolDefinitions } from "../pi-tool-definition-adapter.js";
 type AnyAgentTool = AgentTool;
 
 export function splitSdkTools(options: { tools: AnyAgentTool[]; sandboxEnabled: boolean }): {
-  builtInTools: string[];
+  builtInTools: NonNullable<CreateAgentSessionOptions["tools"]>;
   customTools: ReturnType<typeof toToolDefinitions>;
 } {
   const { tools } = options;