diff --git a/tests/test_litellm/test_pydantic_validation.py b/tests/test_litellm/test_pydantic_validation.py
index 5be0a70288d..1148217ce9a 100644
--- a/tests/test_litellm/test_pydantic_validation.py
+++ b/tests/test_litellm/test_pydantic_validation.py
@@ -5,7 +5,10 @@ import pytest
 from pydantic import BaseModel, field_validator
 
 import litellm
-from litellm.llms.base_llm.base_utils import type_to_response_format_param
+from litellm.llms.base_llm.base_utils import (
+    _pydantic_model_json_schema,
+    type_to_response_format_param,
+)
 from litellm.types.utils import LlmProviders, ModelResponse
 from litellm.utils import (
     ProviderConfigManager,
@@ -13,7 +16,6 @@ from litellm.utils import (
     _apply_response_format_validation,
     _is_basemodel_class,
     _is_pydantic_basemodel_type,
-    _pydantic_model_json_schema,
     _should_preserve_pydantic_response_format,
     normalize_completion_response_format,
     post_call_processing,