diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py
index 7790596e1..54b362a6d 100644
--- a/tests/basic/test_coder.py
+++ b/tests/basic/test_coder.py
@@ -750,7 +750,7 @@ two
     def test_suggest_shell_commands(self):
         with GitTemporaryDirectory():
             io = InputOutput(yes=True)
-            coder = Coder.create(self.GPT35, None, io=io)
+            coder = Coder.create(self.GPT35, "diff", io=io)
 
             def mock_send(*args, **kwargs):
                 coder.partial_response_content = """Here's a shell command to run:
@@ -765,6 +765,9 @@ This command will print 'Hello, World!' to the console."""
 
             coder.send = mock_send
 
+            # Mock the handle_shell_commands method to check if it's called
+            coder.handle_shell_commands = MagicMock()
+
             # Run the coder with a message
             coder.run(with_message="Suggest a shell command")
 
@@ -772,12 +775,6 @@ This command will print 'Hello, World!' to the console."""
             self.assertEqual(len(coder.shell_commands), 1)
             self.assertEqual(coder.shell_commands[0], 'echo "Hello, World!"')
 
-            # Mock the handle_shell_commands method to check if it's called
-            coder.handle_shell_commands = MagicMock()
-
-            # Run the coder again to trigger the shell command execution
-            coder.run(with_message="Run the suggested command")
-
             # Check if handle_shell_commands was called with the correct argument
             coder.handle_shell_commands.assert_called_once_with('echo "Hello, World!"', ANY)