diff --git a/test_tipc/test_inference_jeston.sh b/test_tipc/test_inference_jeston.sh
index c8d0d7d6a9..5d89f1829c 100644
--- a/test_tipc/test_inference_jeston.sh
+++ b/test_tipc/test_inference_jeston.sh
@@ -125,7 +125,7 @@ function func_inference(){
 
 
 
-if [ ${MODE} = "klquant_whole_infer" ]; then
+if [ ${MODE} = "whole_infer" ]; then
     GPUID=$3
     if [ ${#GPUID} -le 0 ];then
         env=" "