diff --git a/script/get-dataset-waymo/run.sh b/script/get-dataset-waymo/run.sh
index fa7001aad..0af8d4a79 100644
--- a/script/get-dataset-waymo/run.sh
+++ b/script/get-dataset-waymo/run.sh
@@ -7,8 +7,8 @@
 
 #${MLC_PYTHON_BIN_WITH_PATH} contains the path to python binary if "get,python" is added as a dependency
 
-if [[ "$MLC_DOWNLOAD_MODE" != "dry" && "$MLC_TMP_REQUIRE_DOWNLOAD" = "true" ]]; then
-  cd "${MLC_DATASET_WAYMO_PATH}/kitti_format/training" || exit
+if [[ "$MLC_DOWNLOAD_MODE" != "dry" && "$MLC_TMP_REQUIRE_DOWNLOAD" = "yes" ]]; then
+  cd "${MLC_DATASET_WAYMO_PATH}/training" || exit
   for f in *.tar.gz; do tar -xzvf "$f"; done
   cd - || exit
 fi
\ No newline at end of file