diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local
index 3affaad67d..ee4992fcff 100644
--- a/docker/local/Dockerfile.local
+++ b/docker/local/Dockerfile.local
@@ -53,6 +53,7 @@ ARG TRANSFORMERS_VERSION=4.28.1
 RUN cd /root; git clone --branch v${DGL_VERSION} --single-branch https://github.com/dmlc/dgl.git
 ENV DGL_HOME=/root/dgl
 ENV DGLBACKEND=pytorch
+ENV PYTHONPATH="/root/dgl/tools/:${PYTHONPATH}"
 
 FROM base as base-cpu
 
@@ -139,9 +140,6 @@ RUN mkdir -p ${SSHDIR} \
 
 EXPOSE ${SSH_PORT}
 
-ENV PYTHONPATH="/root/dgl/tools/:${PYTHONPATH}"
-
-
 # Copy GraphStorm source and add to PYTHONPATH
 RUN mkdir -p /graphstorm
 COPY code/python/graphstorm /graphstorm/python/graphstorm