diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 7df75bc99..bf8e0ba71 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -377,7 +377,7 @@ class TestMain(TestCase):
 
     def test_map_tokens_option(self):
         with GitTemporaryDirectory():
-            with patch("aider.repomap.RepoMap") as MockRepoMap:
+            with patch("aider.coders.base_coder.RepoMap") as MockRepoMap:
                 main(
                     ["--model", "gpt-4", "--map-tokens", "0", "--exit", "--yes"],
                     input=DummyInput(),
@@ -387,7 +387,7 @@ class TestMain(TestCase):
 
     def test_map_tokens_option_with_non_zero_value(self):
         with GitTemporaryDirectory():
-            with patch("aider.repomap.RepoMap") as MockRepoMap:
+            with patch("aider.coders.base_coder.RepoMap") as MockRepoMap:
                 MockRepoMap.return_value.max_map_tokens = 1000
                 main(
                     ["--model", "gpt-4", "--map-tokens", "1000", "--exit", "--yes"],