diff --git a/litellm/main.py b/litellm/main.py
index 2fef048a698..407e851f5b6 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -366,7 +366,7 @@ def completion(
     n: Optional[int] = None,
     stream: Optional[bool] = None,
     stop=None,
-    max_tokens: Optional[float] = None,
+    max_tokens: Optional[int] = None,
     presence_penalty: Optional[float] = None,
     frequency_penalty: Optional[float] = None,
     logit_bias: Optional[dict] = None,