diff --git a/src/tools/call-omo-agent/tools.test.ts b/src/tools/call-omo-agent/tools.test.ts
index 6bbadfa28..455e2f9aa 100644
--- a/src/tools/call-omo-agent/tools.test.ts
+++ b/src/tools/call-omo-agent/tools.test.ts
@@ -298,7 +298,7 @@ describe("createCallOmoAgent", () => {
       getTask: mock(() => undefined),
     }
     const toolDef = createCallOmoAgent(
-      mockCtx,
+      createMockCtx(DEFAULT_AGENTS),
       managerWithLaunch,
       [],
       {
@@ -347,7 +347,7 @@ describe("createCallOmoAgent", () => {
       getTask: mock(() => undefined),
     }
     const toolDef = createCallOmoAgent(
-      mockCtx,
+      createMockCtx(DEFAULT_AGENTS),
       managerWithLaunch,
       [],
       {
@@ -398,7 +398,7 @@ describe("createCallOmoAgent", () => {
       getTask: mock(() => undefined),
     }
     const toolDef = createCallOmoAgent(
-      mockCtx,
+      createMockCtx(DEFAULT_AGENTS),
       managerWithLaunch,
       [],
       {
@@ -448,7 +448,7 @@ describe("createCallOmoAgent", () => {
       getTask: mock(() => undefined),
     }
     const toolDef = createCallOmoAgent(
-      mockCtx,
+      createMockCtx(DEFAULT_AGENTS),
       managerWithLaunch,
       [],
       {