diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 7aeb5d3c4..3605bd72e 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -999,7 +999,7 @@ class TestMain(TestCase):
             coder = main(
                 ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True
             )
-            self.assertIn("flash", coder.main_model.name.lower())
+            self.assertIn("gemini", coder.main_model.name.lower())
             del os.environ["GEMINI_API_KEY"]
 
             # Test no API keys