diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
index ee9b341..fa7e3a0 100644
--- a/tests/unit/test_client.py
+++ b/tests/unit/test_client.py
@@ -219,13 +219,22 @@ def test_request_limit_then_success(self, client, rate_limit_response):
                 APIException(code="conflict", message="Error", details=None),
                 True,
             ),
-            (APIException(code=409, message="Conflict", details=None), False),
-            (APIException(code=502, message="Bad Gateway", details=None), True),
+            (
+                APIException(code=409, message="Conflict", details=None),
+                False,
+            ),
+            (
+                APIException(code=502, message="Bad Gateway", details=None),
+                True,
+            ),
             (
                 APIException(code=503, message="Service Unavailable", details=None),
                 False,
             ),
-            (APIException(code=504, message="Gateway Timeout", details=None), True),
+            (
+                APIException(code=504, message="Gateway Timeout", details=None),
+                True,
+            ),
         ],
     )
     def test_retry_policy(self, client, exception, expected):