diff --git a/prepare_data/prepare_training_data.py b/prepare_data/prepare_training_data.py
index 71cb917..3d899e3 100644
--- a/prepare_data/prepare_training_data.py
+++ b/prepare_data/prepare_training_data.py
@@ -74,7 +74,7 @@ def main(args):
 
     csv_file = target_dir / f"{args.dataset_name}.csv"
     COLUMNS = [
-        'videoid', 'contentUrl', 'duration', 'data_dir', 'name',
+        'videoid', 'contentUrl', 'duration', 'data_dir', 'instruction',
         'dynamic_confidence', 'dynamic_wording', 'dynamic_source_category',
         'embodiment'
     ]