diff --git a/litellm/tests/test_stream_chunk_builder.py b/litellm/tests/test_stream_chunk_builder.py
index a0ae18d6cc..492c09f8ca 100644
--- a/litellm/tests/test_stream_chunk_builder.py
+++ b/litellm/tests/test_stream_chunk_builder.py
@@ -1,4 +1,4 @@
-from litellm import completion
+from litellm import completion, stream_chunk_builder
 import litellm
 import os