diff --git a/scripts/kernelllm.py b/scripts/kernelllm.py
index 68e58dc..85fcc6e 100644
--- a/scripts/kernelllm.py
+++ b/scripts/kernelllm.py
@@ -318,7 +318,7 @@ def run_repl(self):
             if not code:
                 print(f"Using default prompt:\n{DEFAULT_MODEL_CODE}\n")
                 code = DEFAULT_MODEL_CODE
-            prompt = PROMPT_TEMPLATE.format(DEFAULT_MODEL_CODE)
+            prompt = PROMPT_TEMPLATE.format(code)
 
             try:
                 self.stream_raw(prompt)