diff --git a/bigcode_eval/generation.py b/bigcode_eval/generation.py
index 98e15a7be..e44e69e79 100644
--- a/bigcode_eval/generation.py
+++ b/bigcode_eval/generation.py
@@ -70,7 +70,7 @@ def parallel_generations(
         "temperature": args.temperature,
         "top_p": args.top_p,
         "top_k": args.top_k,
-        "max_length": args.max_length_generation,
+        "max_new_tokens": args.max_length_generation,
     }
     stopping_criteria = []
     # The input_length / start_length set to 0 for now will be adjusted later