diff --git a/packages/cli/src/gemini.test.tsx b/packages/cli/src/gemini.test.tsx
index 41f9978d7c..bd8f361e7c 100644
--- a/packages/cli/src/gemini.test.tsx
+++ b/packages/cli/src/gemini.test.tsx
@@ -538,6 +538,7 @@ describe('gemini.tsx main function kitty protocol', () => {
 
     const mockConfig = createMockConfig({
       isInteractive: () => false,
+      getAllowedTools: vi.fn(),
       getQuestion: () => '',
       getSandbox: () => undefined,
       getListExtensions: () => flag === 'listExtensions',
@@ -614,6 +615,7 @@ describe('gemini.tsx main function kitty protocol', () => {
 
     const mockConfig = createMockConfig({
       isInteractive: () => false,
+      getAllowedTools: vi.fn(),
       getQuestion: () => '',
       getSandbox: () => ({ command: 'docker', image: 'test-image' }),
     });
diff --git a/packages/cli/src/gemini_cleanup.test.tsx b/packages/cli/src/gemini_cleanup.test.tsx
index 17e3380f2c..6bc3d46264 100644
--- a/packages/cli/src/gemini_cleanup.test.tsx
+++ b/packages/cli/src/gemini_cleanup.test.tsx
@@ -78,6 +78,7 @@ vi.mock('./config/config.js', () => ({
     getQuestion: vi.fn(() => ''),
     isInteractive: () => false,
     storage: { initialize: vi.fn().mockResolvedValue(undefined) },
+    getAllowedTools: vi.fn(),
   } as unknown as Config),
   parseArguments: vi.fn().mockResolvedValue({}),
   isDebugMode: vi.fn(() => false),
@@ -188,6 +189,7 @@ describe('gemini.tsx main function cleanup', () => {
     } as any); // eslint-disable-line @typescript-eslint/no-explicit-any
     vi.mocked(loadCliConfig).mockResolvedValue({
       isInteractive: vi.fn(() => false),
+      getAllowedTools: vi.fn(),
       getQuestion: vi.fn(() => 'test'),
       getSandbox: vi.fn(() => false),
       getDebugMode: vi.fn(() => false),