diff --git a/tests/test_generative_models.py b/tests/test_generative_models.py
index 637308310..74469e5b8 100644
--- a/tests/test_generative_models.py
+++ b/tests/test_generative_models.py
@@ -811,7 +811,9 @@ def test_tool_config(self, tool_config, expected_tool_config):
     )
     def test_system_instruction(self, instruction, expected_instr):
         self.responses["generate_content"] = [simple_response("echo echo")]
-        model = generative_models.GenerativeModel("gemini-1.5-flash", system_instruction=instruction)
+        model = generative_models.GenerativeModel(
+            "gemini-1.5-flash", system_instruction=instruction
+        )
 
         _ = model.generate_content("test")
 
@@ -1206,7 +1208,9 @@ def test_repr_for_broken_streaming_chat(self):
         self.assertEqual(expected, result)
 
     def test_repr_for_system_instruction(self):
-        model = generative_models.GenerativeModel("gemini-1.5-flash", system_instruction="Be excellent.")
+        model = generative_models.GenerativeModel(
+            "gemini-1.5-flash", system_instruction="Be excellent."
+        )
         result = repr(model)
         self.assertIn("system_instruction='Be excellent.'", result)