diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py
index b8ebc05bb..aa29b509d 100644
--- a/litellm/llms/sagemaker.py
+++ b/litellm/llms/sagemaker.py
@@ -30,7 +30,7 @@ import json
 
 
 class TokenIterator:
-    def __init__(self, stream, acompletion: bool):
+    def __init__(self, stream, acompletion: bool = False):
         if acompletion == False:
             self.byte_iterator = iter(stream)
         elif acompletion == True: