diff --git a/code/image_classfication/run_vanilla_kd.sh b/code/image_classfication/run_vanilla_kd.sh
index 3c402b8..1dafa3a 100755
--- a/code/image_classfication/run_vanilla_kd.sh
+++ b/code/image_classfication/run_vanilla_kd.sh
@@ -5,7 +5,7 @@ do
     python main.py --device=cuda:0 \
                    --teacher=wrn-28-4 \
                    --student=wrn-28-4 \
-                   --dataset=cifar100 \
+                   --dataset=cifar10 \
                    --batch_size=128 \
                    --num_epochs=200 \
                    --mode=crop \