diff --git a/tests/test_litellm/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py b/tests/test_litellm/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py
index 06647b2b66b..6af766743cf 100644
--- a/tests/test_litellm/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py
+++ b/tests/test_litellm/completion_extras/litellm_responses_transformation/test_completion_extras_litellm_responses_transformation_transformation.py
@@ -1,8 +1,9 @@
 import json
 import os
 import sys
-from unittest.mock import Mock
+from unittest.mock import MagicMock, Mock, patch
 
+import httpx
 import pytest
 
 sys.path.insert(