diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 6adecb418..28824470b 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -378,11 +378,19 @@ class TestMain(TestCase):
     def test_map_tokens_option(self):
         with GitTemporaryDirectory():
             with patch("aider.repomap.RepoMap") as MockRepoMap:
-                main(["--model", "gpt-4", "--map-tokens", "0", "--exit"], input=DummyInput(), output=DummyOutput())
+                main(
+                    ["--model", "gpt-4", "--map-tokens", "0", "--exit"],
+                    input=DummyInput(),
+                    output=DummyOutput(),
+                )
                 MockRepoMap.assert_not_called()
 
     def test_map_tokens_option_with_non_zero_value(self):
         with GitTemporaryDirectory():
             with patch("aider.repomap.RepoMap") as MockRepoMap:
-                main(["--model", "gpt-4", "--map-tokens", "1000", "--exit"], input=DummyInput(), output=DummyOutput())
+                main(
+                    ["--model", "gpt-4", "--map-tokens", "1000", "--exit"],
+                    input=DummyInput(),
+                    output=DummyOutput(),
+                )
                 MockRepoMap.assert_called_once()