diff --git a/prompt-learning/prompt_utils.py b/prompt-learning/prompt_utils.py
index cb5061e..c9564ca 100644
--- a/prompt-learning/prompt_utils.py
+++ b/prompt-learning/prompt_utils.py
@@ -17,7 +17,7 @@ PROMPT_TEMPLATES = {
         "generate": "belief states: value = $value, slot ="
     },
     "inverse-prompt": {
-        "training": INVERSE_PROMPTS["i1"],
+        "training": INVERSE_PROMPTS["i2"],
     },
     "prompt-ensemble": {
         "training": {