diff --git a/tools/serving_build.sh b/tools/serving_build.sh
index 6479aaf00f51147386a40b3e0c6f14ad6b7c2584..96c8c3657d73b0f9717b3761a9983c2641377d2e 100644
--- a/tools/serving_build.sh
+++ b/tools/serving_build.sh
@@ -706,7 +706,7 @@ function python_test_pipeline(){
             sleep 5
             
             # test: thread servicer & thread op
-            cat << EOF > config.yaml
+            cat << EOF > config.yml
 port: 18080
 worker_num: 2
 build_dag_each_worker: false
@@ -722,7 +722,7 @@ EOF
             ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
 
             # test: thread servicer & process op
-            cat << EOF > config.yaml
+            cat << EOF > config.yml
 port: 18080
 worker_num: 2
 build_dag_each_worker: false
@@ -738,7 +738,7 @@ EOF
             ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
 
             # test: process servicer & thread op
-            cat << EOF > config.yaml
+            cat << EOF > config.yml
 port: 18080
 worker_num: 2
 build_dag_each_worker: true
@@ -754,7 +754,7 @@ EOF
             ps -ef | grep "pipeline_server" | grep -v grep | awk '{print $2}' | xargs kill
 
             # test: process servicer & process op
-            cat << EOF > config.yaml
+            cat << EOF > config.yml
 port: 18080
 worker_num: 2
 build_dag_each_worker: false