diff --git a/.devcontainer/non-root-user/Dockerfile b/.devcontainer/non-root-user/Dockerfile
index c7052577aae266e2d508b69fc26105745885844f..2ea277ce0be831ae6cfb9cce0469ca91f90b7a27 100644
--- a/.devcontainer/non-root-user/Dockerfile
+++ b/.devcontainer/non-root-user/Dockerfile
@@ -36,7 +36,8 @@ RUN groupadd --gid $USER_GID $USERNAME && \
 
 USER $USERNAME
 
-RUN export REMOTE="https://gitee.com/mirrors/oh-my-zsh" && \
+RUN echo 'source ~/venv/bin/activate' >> ~/.zshrc && \
+    export REMOTE="https://gitee.com/mirrors/oh-my-zsh" && \
     export COMMAND="$(curl -fsSL ${REMOTE}/raw/master/tools/install.sh)" && \
     sh -c "${COMMAND}" "" --unattended
     # echo 'export PATH="/usr/local/cuda-12.4/bin${PATH:+:${PATH}}"' >>~/.zshrc && \
@@ -50,5 +51,3 @@ RUN source ~/venv/bin/activate && \
     pip install --upgrade pip && \
     pip install --no-cache-dir -r ~/requirements-dev.txt && \
     rm -rf ~/requirements.txt ~/requirements-dev.txt
-
-RUN echo 'source ~/venv/bin/activate' >> ~/.zshrc