diff --git a/bigcodebench/provider/utility.py b/bigcodebench/provider/utility.py
index 9c5b48f..ad09159 100644
--- a/bigcodebench/provider/utility.py
+++ b/bigcodebench/provider/utility.py
@@ -72,7 +72,7 @@ def make_raw_chat_prompt(
                 [
                     {"role": "user", "content": task_prompt},
                 ],
-                tokenize=False,
+                tokenize=False, add_generation_prompt=True
             ).split(_MAGIC_SPLITTER_)[0]
     return task_prompt