diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 28824470b..d7f8ee952 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -379,7 +379,7 @@ class TestMain(TestCase):
         with GitTemporaryDirectory():
             with patch("aider.repomap.RepoMap") as MockRepoMap:
                 main(
-                    ["--model", "gpt-4", "--map-tokens", "0", "--exit"],
+                    ["--model", "gpt-4", "--map-tokens", "0", "--exit", "--yes"],
                     input=DummyInput(),
                     output=DummyOutput(),
                 )
@@ -389,7 +389,7 @@ class TestMain(TestCase):
         with GitTemporaryDirectory():
             with patch("aider.repomap.RepoMap") as MockRepoMap:
                 main(
-                    ["--model", "gpt-4", "--map-tokens", "1000", "--exit"],
+                    ["--model", "gpt-4", "--map-tokens", "1000", "--exit", "--yes"],
                     input=DummyInput(),
                     output=DummyOutput(),
                 )