diff --git a/training/configs/ornith-35b-lora.yaml b/training/configs/ornith-35b-lora.yaml
index 2cfb555..17c6fab 100644
--- a/training/configs/ornith-35b-lora.yaml
+++ b/training/configs/ornith-35b-lora.yaml
@@ -26,7 +26,7 @@ lora_task_type: CAUSAL_LM
 dataset:
   - path: /home/cyaren/loras/agenx-lora-training/dataset/combined_20k.jsonl
     type: completion
-    text_column: text
+    text_column: output
 
 # Training Parameters
 train_params: