diff --git a/litellm/tests/test_stream_chunk_builder.py b/litellm/tests/test_stream_chunk_builder.py
index fbfc78554a..a0ae18d6cc 100644
--- a/litellm/tests/test_stream_chunk_builder.py
+++ b/litellm/tests/test_stream_chunk_builder.py
@@ -10,7 +10,8 @@ def test_stream_chunk_builder():
     response = completion(
         model="gpt-3.5-turbo",
         messages=messages,
-        stream=True
+        stream=True,
+        max_tokens=10,
     )
 
     for chunk in response: