1 files changed, 4 insertions, 0 deletions
diff --git a/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx b/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx
index f3c0764e..38400caf 100644
--- a/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx
+++ b/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx
@@ -26,7 +26,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => {
       sessionStartTime: new Date(),
       metrics,
       lastPromptTokenCount: 0,
+      promptCount: 5,
     },
+
+    getPromptCount: () => 5,
+    startNewPrompt: vi.fn(),
   });
 
   return render(<SessionSummaryDisplay duration="1h 23m 45s" />);