diff --git a/litellm/llms/xai/chat/transformation.py b/litellm/llms/xai/chat/transformation.py
index 8a42f196eef..351da445f08 100644
--- a/litellm/llms/xai/chat/transformation.py
+++ b/litellm/llms/xai/chat/transformation.py
@@ -118,7 +118,10 @@ class XAIChatConfig(OpenAIGPTConfig):
         )
 
     def get_error_class(
-        self, error_message: str, status_code: int, headers: dict | httpx.Headers  # mutable-ok: mirrors the base class signature
+        self,
+        error_message: str,
+        status_code: int,
+        headers: dict | httpx.Headers,  # mutable-ok: mirrors the base class signature
     ) -> BaseLLMException:
         return super().get_error_class(
             error_message=error_message,
diff --git a/litellm/llms/xai/responses/transformation.py b/litellm/llms/xai/responses/transformation.py
index af9866479c8..68d2f839f86 100644
--- a/litellm/llms/xai/responses/transformation.py
+++ b/litellm/llms/xai/responses/transformation.py
@@ -256,7 +256,10 @@ class XAIResponsesAPIConfig(OpenAIResponsesAPIConfig):
         return f"{api_base}/responses"
 
     def get_error_class(
-        self, error_message: str, status_code: int, headers: dict | httpx.Headers  # mutable-ok: mirrors the base class signature
+        self,
+        error_message: str,
+        status_code: int,
+        headers: dict | httpx.Headers,  # mutable-ok: mirrors the base class signature
     ) -> BaseLLMException:
         return super().get_error_class(
             error_message=error_message,