diff --git a/script/build-dockerfile/customize.py b/script/build-dockerfile/customize.py
index fe680c3b1..23e7891dc 100644
--- a/script/build-dockerfile/customize.py
+++ b/script/build-dockerfile/customize.py
@@ -297,6 +297,8 @@ def preprocess(i):
         f.write('ENV PATH="$HOME/venv/mlc/bin:$PATH"' + EOL)
     # f.write('RUN . /opt/venv/mlc/bin/activate' + EOL)
 
+    f.write('ENV PATH="$PATH:$HOME/.local/bin"' + EOL)
+
     f.write(
         'RUN {} -m pip install '.format(python) +
         " ".join(
diff --git a/script/build-dockerfile/dockerinfo.json b/script/build-dockerfile/dockerinfo.json
index a189267a8..cfb739e77 100644
--- a/script/build-dockerfile/dockerinfo.json
+++ b/script/build-dockerfile/dockerinfo.json
@@ -11,8 +11,7 @@
   },
   "ENTRYPOINT": "[\"/bin/bash\", \"-c\"]",
   "ENV": {
-    "TZ": "US/Pacific",
-    "PATH": "${PATH}:/home/mlcuser/.local/bin"
+    "TZ": "US/Pacific"
   },
   "RUN_CMDS": [
     "ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ >/etc/timezone"
diff --git a/script/run-mlperf-automotive-app/customize.py b/script/run-mlperf-automotive-app/customize.py
index d9ca6f3e0..6f8398743 100644
--- a/script/run-mlperf-automotive-app/customize.py
+++ b/script/run-mlperf-automotive-app/customize.py
@@ -3,7 +3,6 @@
 import json
 import shutil
 import subprocess
-import mlc
 import copy
 from tabulate import tabulate
 
@@ -20,6 +19,7 @@ def preprocess(i):
     inp = i['input']
     state = i['state']
     script_path = i['run_script_input']['path']
+    mlc = i['automation'].action_object
 
     if env.get('MLC_RUN_DOCKER_CONTAINER', '') == "yes":
         return {'return': 0}