diff --git a/base.dockerfile b/base.dockerfile
index c1db6d6..e2c82a6 100644
--- a/base.dockerfile
+++ b/base.dockerfile
@@ -86,7 +86,7 @@ EOF
 # Change to non-root user
 USER rocm
 WORKDIR /home/rocm
-ENV PATH="${PATH}:/opt/rocm/bin"
+ENV PATH="${PATH}:/opt/rocm/bin:/home/rocm/.local/bin"
 
 # Default to login shell
 CMD [ "bash", "-l" ]
diff --git a/tensorflow.dockerfile b/tensorflow.dockerfile
index be61f3e..1db4bbe 100644
--- a/tensorflow.dockerfile
+++ b/tensorflow.dockerfile
@@ -32,8 +32,6 @@ sudo apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev
 sudo rm -rf /var/lib/apt/lists/*
 EOF
 
-ENV PATH=$PATH:/home/rocm/.local/bin
-
 # TODO: simplify dependencies
 # > Ref: https://github.com/ROCm/ROCm/issues/1880
 RUN python${PYTHON_VERSION} -m pip install -U --user pip numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf