diff --git a/packages/cli/src/ui/components/ToastDisplay.test.tsx b/packages/cli/src/ui/components/ToastDisplay.test.tsx
index 477fa47f62..06baa00da9 100644
--- a/packages/cli/src/ui/components/ToastDisplay.test.tsx
+++ b/packages/cli/src/ui/components/ToastDisplay.test.tsx
@@ -16,6 +16,7 @@ import { type HistoryItem } from '../types.js';
 const renderToastDisplay = async (
   uiState: Partial<UIState> = {},
   inputState: Partial<InputState> = {},
+  allowEmptyFrame = false,
 ) =>
   renderWithProviders(<ToastDisplay />, {
     uiState: {
@@ -27,6 +28,7 @@ const renderToastDisplay = async (
       showEscapePrompt: false,
       ...inputState,
     },
+    allowEmptyFrame,
   });
 
 describe('ToastDisplay', () => {
@@ -155,7 +157,7 @@ describe('ToastDisplay', () => {
   });
 
   it('renders nothing by default', async () => {
-    const { lastFrame } = await renderToastDisplay();
+    const { lastFrame } = await renderToastDisplay({}, {}, true);
     expect(lastFrame({ allowEmpty: true })).toBe('');
   });