diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index a53ce67a5..9fe8a9f21 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -1277,7 +1277,7 @@ class TestMain(TestCase):
         mock_io_instance = MockInputOutput.return_value
         with GitTemporaryDirectory():
             main(
-                ["--cache-prompts", "--exit", "--yes"],
+                ["--cache-prompts", "--exit", "--yes", "--no-stream"],
                 input=DummyInput(),
                 output=DummyOutput(),
             )