diff --git a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
index f34b36d3d..f0fcb1ef9 100644
--- a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
+++ b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
@@ -3323,6 +3323,7 @@ describe('useGeminiStream', () => {
                 'Reason 2: More details needed',
                 'Reason 3: Incomplete response',
               ],
+              stopHookCount: 3,
             },
           };
         })(),
@@ -3365,6 +3366,7 @@ describe('useGeminiStream', () => {
             value: {
               iterationCount: 5,
               reasons: ['Hook reason 1', 'Hook reason 2'],
+              stopHookCount: 2,
             },
           };
         })(),