diff --git a/scripts/apollo_buildify.sh b/scripts/apollo_buildify.sh
index 00080715060f945973002d6064d9975862960320..d58ae5e65a686ae04a111172e6d4576a2ee05eae 100755
--- a/scripts/apollo_buildify.sh
+++ b/scripts/apollo_buildify.sh
@@ -20,7 +20,7 @@ function buildify() {
     buildifier_cmd="${buildifier_cmd} -mode=fix"
 
     local build_dirs="cyber third_party tools external"
-    if [ "${STAGE}" == "dev" ]; then
+    if [ "${stage}" == "dev" ]; then
         build_dirs="modules ${build_dirs}"
     fi
     build_dirs=$(printf "${APOLLO_ROOT_DIR}/%s " $build_dirs)