diff --git a/src/distilabel/models/llms/azure.py b/src/distilabel/models/llms/azure.py
index b9132991a..6b3fd9076 100644
--- a/src/distilabel/models/llms/azure.py
+++ b/src/distilabel/models/llms/azure.py
@@ -122,7 +122,8 @@ def load(self) -> None:
         # This is a workaround to avoid the `OpenAILLM` calling the _prepare_structured_output
         # in the load method before we have the proper client.
         with patch(
-            "distilabel.models.openai.OpenAILLM._prepare_structured_output", lambda x: x
+            "distilabel.models.llms.openai.OpenAILLM._prepare_structured_output",
+            lambda x: x,
         ):
             super().load()