1 files changed, 4 insertions, 11 deletions
diff --git a/packages/cli/src/ui/hooks/useToolScheduler.test.ts b/packages/cli/src/ui/hooks/useToolScheduler.test.ts
index c5d968fe..b2071931 100644
--- a/packages/cli/src/ui/hooks/useToolScheduler.test.ts
+++ b/packages/cli/src/ui/hooks/useToolScheduler.test.ts
@@ -36,13 +36,6 @@ import {
   HistoryItemToolGroup,
 } from '../types.js';
 
-vi.mock('./useTerminalSize', () => ({
-  useTerminalSize: () => ({
-    columns: 80,
-    rows: 24,
-  }),
-}));
-
 // Mocks
 vi.mock('@google/gemini-cli-core', async () => {
   const actual = await vi.importActual('@google/gemini-cli-core');
@@ -231,8 +224,8 @@ describe('useReactToolScheduler in YOLO Mode', () => {
       request.args,
       expect.any(AbortSignal),
       undefined,
-      80,
-      24,
+      undefined,
+      undefined,
     );
 
     // Check that onComplete was called with success
@@ -383,8 +376,8 @@ describe('useReactToolScheduler', () => {
       request.args,
       expect.any(AbortSignal),
       undefined,
-      80,
-      24,
+      undefined,
+      undefined,
     );
     expect(onComplete).toHaveBeenCalledWith([
       expect.objectContaining({