diff --git a/tests/test_main.py b/tests/test_main.py
index 097063208..a684fe368 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -18,7 +18,7 @@ class TestMain(TestCase):
         self.original_cwd = os.getcwd()
         self.tempdir = tempfile.mkdtemp()
         os.chdir(self.tempdir)
-        self.patcher = patch("aider.main.Coder.check_model_availability")
+        self.patcher = patch("aider.coders.base.check_model_availability")
         self.mock_check = self.patcher.start()
         self.mock_check.return_value = True
 
@@ -48,39 +48,39 @@ class TestMain(TestCase):
         self.assertTrue(os.path.exists("foo.txt"))
 
     def test_main_args(self):
-        with patch("aider.main.Coder") as MockCoder:
+        with patch("aider.main.Coder.create") as MockCoder:
             main(["--no-auto-commits"])
             _, kwargs = MockCoder.call_args
             assert kwargs["auto_commits"] is False
 
-        with patch("aider.main.Coder") as MockCoder:
+        with patch("aider.main.Coder.create") as MockCoder:
             main(["--auto-commits"])
             _, kwargs = MockCoder.call_args
             assert kwargs["auto_commits"] is True
 
-        with patch("aider.main.Coder") as MockCoder:
+        with patch("aider.main.Coder.create") as MockCoder:
             main([])
             _, kwargs = MockCoder.call_args
             assert kwargs["dirty_commits"] is True
             assert kwargs["auto_commits"] is True
             assert kwargs["pretty"] is True
 
-        with patch("aider.main.Coder") as MockCoder:
+        with patch("aider.main.Coder.create") as MockCoder:
             main(["--no-pretty"])
             _, kwargs = MockCoder.call_args
             assert kwargs["pretty"] is False
 
-        with patch("aider.main.Coder") as MockCoder:
+        with patch("aider.main.Coder.create") as MockCoder:
             main(["--pretty"])
             _, kwargs = MockCoder.call_args
             assert kwargs["pretty"] is True
 
-        with patch("aider.main.Coder") as MockCoder:
+        with patch("aider.main.Coder.create") as MockCoder:
             main(["--no-dirty-commits"])
             _, kwargs = MockCoder.call_args
             assert kwargs["dirty_commits"] is False
 
-        with patch("aider.main.Coder") as MockCoder:
+        with patch("aider.main.Coder.create") as MockCoder:
             main(["--dirty-commits"])
             _, kwargs = MockCoder.call_args
             assert kwargs["dirty_commits"] is True