diff --git a/packages/cli/src/nonInteractiveCli.test.ts b/packages/cli/src/nonInteractiveCli.test.ts
index 3fc827d497..e83e1e60c9 100644
--- a/packages/cli/src/nonInteractiveCli.test.ts
+++ b/packages/cli/src/nonInteractiveCli.test.ts
@@ -300,9 +300,7 @@ describe('runNonInteractive', () => {
       prompt_id: 'prompt-id-stream',
     });
 
-    expect(streamSpy).toHaveBeenCalledWith({
-      streamId: expect.any(String),
-    });
+    expect(streamSpy).toHaveBeenCalledWith({ streamId: expect.any(String) });
   });
 
   it('should register activity logger when GEMINI_CLI_ACTIVITY_LOG_TARGET is set', async () => {
diff --git a/packages/cli/src/nonInteractiveCli.ts b/packages/cli/src/nonInteractiveCli.ts
index e6a85a4886..9b88011058 100644
--- a/packages/cli/src/nonInteractiveCli.ts
+++ b/packages/cli/src/nonInteractiveCli.ts
@@ -466,7 +466,7 @@ export async function runNonInteractive({
             }
             break;
           }
-          case 'stream_end': {
+          case 'agent_end': {
             if (event.reason === 'aborted') {
               handleCancellationError(config);
             } else if (event.reason === 'max_turns') {