diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 823a89391..7f494f55d 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -749,31 +749,41 @@ class TestMain(TestCase):
         with GitTemporaryDirectory():
             # Test Anthropic API key
             os.environ["ANTHROPIC_API_KEY"] = "test-key"
-            coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True)
+            coder = main(
+                ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True
+            )
             self.assertEqual(coder.main_model.name, "sonnet")
             del os.environ["ANTHROPIC_API_KEY"]
 
             # Test DeepSeek API key
             os.environ["DEEPSEEK_API_KEY"] = "test-key"
-            coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True)
+            coder = main(
+                ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True
+            )
             self.assertEqual(coder.main_model.name, "deepseek")
             del os.environ["DEEPSEEK_API_KEY"]
 
             # Test OpenRouter API key
             os.environ["OPENROUTER_API_KEY"] = "test-key"
-            coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True)
+            coder = main(
+                ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True
+            )
             self.assertEqual(coder.main_model.name, "openrouter/anthropic/claude-3.5-sonnet")
             del os.environ["OPENROUTER_API_KEY"]
 
             # Test OpenAI API key
             os.environ["OPENAI_API_KEY"] = "test-key"
-            coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True)
+            coder = main(
+                ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True
+            )
             self.assertEqual(coder.main_model.name, "gpt-4o")
             del os.environ["OPENAI_API_KEY"]
 
             # Test Gemini API key
             os.environ["GEMINI_API_KEY"] = "test-key"
-            coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True)
+            coder = main(
+                ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True
+            )
             self.assertEqual(coder.main_model.name, "flash")
             del os.environ["GEMINI_API_KEY"]
 
@@ -786,7 +796,9 @@ class TestMain(TestCase):
             # Test that earlier API keys take precedence
             os.environ["ANTHROPIC_API_KEY"] = "test-key"
             os.environ["OPENAI_API_KEY"] = "test-key"
-            coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True)
+            coder = main(
+                ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True
+            )
             self.assertEqual(coder.main_model.name, "sonnet")
             del os.environ["ANTHROPIC_API_KEY"]
             del os.environ["OPENAI_API_KEY"]