diff --git a/script/get-dataset-waymo/run-rclone.sh b/script/get-dataset-waymo/run-rclone.sh
index caca3ff98..12aac3377 100644
--- a/script/get-dataset-waymo/run-rclone.sh
+++ b/script/get-dataset-waymo/run-rclone.sh
@@ -1,4 +1,7 @@
 cmd="rclone sync mlc-waymo:waymo_preprocessed_dataset/kitti_format ${MLC_DATASET_WAYMO_PATH} -P"
 echo $cmd
 eval $cmd
-test $? -eq 0 || exit $?
\ No newline at end of file
+test $? -eq 0 || exit $?
+cd ${MLC_DATASET_WAYMO_PATH}/kitti_format/training
+for f in *.tar.gz; do tar -xzvf "$f"; done
+cd -
diff --git a/script/run-mlperf-inference-mobilenet-models/customize.py b/script/run-mlperf-inference-mobilenet-models/customize.py
index fb3238ca2..9601af651 100644
--- a/script/run-mlperf-inference-mobilenet-models/customize.py
+++ b/script/run-mlperf-inference-mobilenet-models/customize.py
@@ -200,6 +200,8 @@ def preprocess(i):
                     if r['return'] > 0:
                         print(r)
                     #    return r
+                    else:
+                        importlib.reload(mlc.action)
 
                 if is_true(env.get('MLC_TEST_ONE_RUN', '')):
                     return {'return': 0}