diff --git a/llms/mlx_lm/chat.py b/llms/mlx_lm/chat.py
index 7795d8d7..5a8245ef 100644
--- a/llms/mlx_lm/chat.py
+++ b/llms/mlx_lm/chat.py
@@ -79,7 +79,7 @@ def main():
             model,
             tokenizer,
             prompt,
-            args.max_tokens,
+            max_tokens=args.max_tokens,
             sampler=make_sampler(args.temp, args.top_p),
             prompt_cache=prompt_cache,
         ):