diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 16f0fcbaa..92afe1688 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -226,9 +226,11 @@ class TestMain(TestCase):
 
     def test_main_exit_calls_version_check(self):
         with GitTemporaryDirectory():
-            with patch("aider.main.check_version") as mock_check_version:
+            with patch("aider.main.check_version") as mock_check_version, \
+                 patch("aider.main.InputOutput") as mock_input_output:
                 main(["--exit"], input=DummyInput(), output=DummyOutput())
                 mock_check_version.assert_called_once()
+                mock_input_output.assert_called_once()
 
     @patch("aider.main.InputOutput")
     @patch("aider.coders.base_coder.Coder.run")