1 files changed, 7 insertions, 1 deletions
diff --git a/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts b/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts
index 49dd9c2a..d5270aba 100644
--- a/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts
+++ b/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts
@@ -104,6 +104,8 @@ describe('useShellCommandProcessor', () => {
   ): ShellExecutionResult => ({
     rawOutput: Buffer.from(overrides.output || ''),
     output: 'Success',
+    stdout: 'Success',
+    stderr: '',
     exitCode: 0,
     signal: null,
     error: null,
@@ -221,6 +223,7 @@ describe('useShellCommandProcessor', () => {
       act(() => {
         mockShellOutputCallback({
           type: 'data',
+          stream: 'stdout',
           chunk: 'hello',
         });
       });
@@ -231,9 +234,12 @@ describe('useShellCommandProcessor', () => {
       // Advance time and send another event to trigger the throttled update
       await act(async () => {
         await vi.advanceTimersByTimeAsync(OUTPUT_UPDATE_INTERVAL_MS + 1);
+      });
+      act(() => {
         mockShellOutputCallback({
           type: 'data',
-          chunk: 'hello world',
+          stream: 'stdout',
+          chunk: ' world',
         });
       });