diff --git a/tests/functional/test_battleship_game_flow.py b/tests/functional/test_battleship_game_flow.py
index f13447d..e063c4d 100644
--- a/tests/functional/test_battleship_game_flow.py
+++ b/tests/functional/test_battleship_game_flow.py
@@ -163,7 +163,9 @@ script_result = {
         mock_metadata["ai_board"][90:92] = ["Destroyer"] * 2  # Destroyer
 
         with patch.object(
-            activity, "execute_processing_script", return_value={"metadata": mock_metadata}
+            activity,
+            "execute_processing_script",
+            return_value={"metadata": mock_metadata},
         ) as mock_exec:
             metadata = {}
             result = activity.execute_processing_script(metadata, setup_script)
diff --git a/tests/unit/test_guarded_ai_functions.py b/tests/unit/test_guarded_ai_functions.py
index 4b58d06..26cd6db 100644
--- a/tests/unit/test_guarded_ai_functions.py
+++ b/tests/unit/test_guarded_ai_functions.py
@@ -311,7 +311,9 @@ class TestGuardedAI(unittest.TestCase):
             # Verify client was created and stored with actual model ID
             mock_get_client.assert_called_with("http://test.com", "test-key")
             self.assertIn("test-model-id", guarded_ai.MODEL_CLIENT_MAP)
-            self.assertEqual(guarded_ai.MODEL_CLIENT_MAP["test-model-id"][0], mock_client)
+            self.assertEqual(
+                guarded_ai.MODEL_CLIENT_MAP["test-model-id"][0], mock_client
+            )
             self.assertEqual(
                 guarded_ai.MODEL_CLIENT_MAP["test-model-id"][1], "http://test.com"
             )