diff --git a/training/amygdala_training/train_direct.py b/training/amygdala_training/train_direct.py
index 02792b3..8749e37 100644
--- a/training/amygdala_training/train_direct.py
+++ b/training/amygdala_training/train_direct.py
@@ -86,7 +86,10 @@ def main() -> None:
 
     def apply_template(text: str) -> str:
         return tokenizer.apply_chat_template(
-            [{"role": "assistant", "content": text}],
+            [
+                {"role": "user", "content": "How do you feel right now?"},
+                {"role": "assistant", "content": text},
+            ],
             tokenize=False,
         )