diff --git a/code/image_classfication/run_vanilla.sh b/code/image_classfication/run_vanilla.sh
index 0a50513..6da2dcc 100755
--- a/code/image_classfication/run_vanilla.sh
+++ b/code/image_classfication/run_vanilla.sh
@@ -4,7 +4,7 @@
 
 python main.py --device=cuda:0 \
                --student=wrn-28-2 \
-               --dataset=cifar100 \
+               --dataset=cifar10 \
                --batch_size=128 \
                --num_epochs=200 \
                --mode=crop \