diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 57c7b6b29b83464ce33769a5debe4083abf15ebe..1b09cff6b7ea9f30c13fc0d64e6c37875375ba05 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -51,7 +51,7 @@ if [ ${MODE} = "lite_train_lite_infer" ] || [ ${MODE} = "lite_train_whole_infer"
 	cp -r train/* val/
     fi
     cd ../../
-elif [ ${MODE} = "whole_infer" ] || [ ${MODE} = "cpp_infer" ];then
+elif [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ];then
     # download data
     cd dataset
     rm -rf ILSVRC2012