diff --git a/litellm/llms/azure_ai/image_generation/flux_transformation.py b/litellm/llms/azure_ai/image_generation/flux_transformation.py
index b10e8a6f35e..b9d5e11ff2a 100644
--- a/litellm/llms/azure_ai/image_generation/flux_transformation.py
+++ b/litellm/llms/azure_ai/image_generation/flux_transformation.py
@@ -102,7 +102,7 @@ class AzureFoundryFluxImageGenerationConfig(GPTImageGenerationConfig):
         optional_params: Mapping[str, object],
         model: str,
         drop_params: bool,
-    ) -> dict:  # mutable-ok: inherited config contract returns a dict
+    ) -> dict[str, object]:  # mutable-ok: inherited config contract returns a dict
         if not self.is_flux2_model(model):
             return super().map_openai_params(
                 non_default_params=dict(non_default_params),