diff --git a/tests/basic/test_commands.py b/tests/basic/test_commands.py
index f2d946b33..0bc7a33e2 100644
--- a/tests/basic/test_commands.py
+++ b/tests/basic/test_commands.py
@@ -1683,7 +1683,7 @@ class TestCommands(TestCase):
             context.exception.kwargs.get("main_model").weak_model.name, self.GPT35.weak_model.name
         )
         # Check that the edit format is updated to the new model's default
-        self.assertEqual(context.exception.kwargs.get("edit_format"), "whole")
+        self.assertEqual(context.exception.kwargs.get("edit_format"), "diff")
 
     def test_cmd_model_preserves_explicit_edit_format(self):
         io = InputOutput(pretty=False, fancy_input=False, yes=True)
@@ -1754,7 +1754,7 @@ class TestCommands(TestCase):
         # Check that the SwitchCoder exception contains the correct model configuration
         self.assertEqual(context.exception.kwargs.get("main_model").name, "gpt-4")
         # Check that the edit format is updated to the new model's default
-        self.assertEqual(context.exception.kwargs.get("edit_format"), "whole")
+        self.assertEqual(context.exception.kwargs.get("edit_format"), "diff")
 
     def test_cmd_ask(self):
         io = InputOutput(pretty=False, fancy_input=False, yes=True)