diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py
index f1d97ab8ebb..41cba5445e3 100644
--- a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py
+++ b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py
@@ -3,6 +3,7 @@
 import os
 from unittest import mock
 
+import httpx
 import pytest
 from langchain_core.messages import HumanMessage
 from typing_extensions import TypedDict
@@ -99,3 +100,4 @@ def test_max_completion_tokens_in_payload() -> None:
         "stream": False,
         "max_completion_tokens": 300,
     }
+