diff --git a/tests/test_litellm/responses/test_streaming_iterator_error_events.py b/tests/test_litellm/responses/test_streaming_iterator_error_events.py
index adaa6b0e3f3..19b69042da1 100644
--- a/tests/test_litellm/responses/test_streaming_iterator_error_events.py
+++ b/tests/test_litellm/responses/test_streaming_iterator_error_events.py
@@ -20,6 +20,7 @@ Pydantic ValidationError (previously typed as Optional[str]).
 """
 
 import json
+from importlib import import_module
 from unittest.mock import Mock, patch
 
 import pytest