diff --git a/scripts/train.sh b/scripts/train.sh
old mode 100644
new mode 100755
index f967ed6..f93af57
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -14,4 +14,4 @@ fi
 
 mkdir -p "$OUTPUT_DIR"
 
-train_gemma --model "google/gemma-3-12b-it" --output-dir "$OUTPUT_DIR"
\ No newline at end of file
+gemma_train --model "google/gemma-3-12b-it" --output-dir "$OUTPUT_DIR"
\ No newline at end of file