diff --git a/run_TI_step_1.sh b/run_TI_step_1.sh
index a2f84bd..18c6df8 100644
--- a/run_TI_step_1.sh
+++ b/run_TI_step_1.sh
@@ -35,7 +35,7 @@ teacher_stochastic=0
 teacher_inverted=0
 
 # Training Type (downstream, qat_normal, qat_step1, qat_step2)
-training_type=qat_normal
+training_type=qat_step1
 
 # DA Options
 aug_train=0
@@ -56,4 +56,4 @@ CUDA_VISIBLE_DEVICES=$1 python /home/ms/workspace/git/Teacher-Intervention-KD-QA
 --neptune ${neptune} \
 --aug_N ${aug_N} \
 --num_train_epochs 3 --seed 1 \
---learning_rate ${learning_rate} 
\ No newline at end of file
+--learning_rate ${learning_rate}