diff --git a/src/promptflow/promptflow/_sdk/data/docker/Dockerfile.jinja2 b/src/promptflow/promptflow/_sdk/data/docker/Dockerfile.jinja2
index 03f1bc12adf..44448a6acca 100644
--- a/src/promptflow/promptflow/_sdk/data/docker/Dockerfile.jinja2
+++ b/src/promptflow/promptflow/_sdk/data/docker/Dockerfile.jinja2
@@ -14,6 +14,9 @@ WORKDIR /
 COPY ./flow/{{env.python_requirements_txt}} /flow/{{env.python_requirements_txt}}
 {% endif %}
 
+# gcc is for build psutil in MacOS
+RUN apt-get update && apt-get install -y runit gcc
+
 # create conda environment
 {% if env.conda_file %}
 COPY ./flow/{{env.conda_file}} /flow/{{env.conda_file}}
@@ -44,8 +47,6 @@ COPY ./flow /flow
 RUN conda run -n {{env.conda_env_name}} sh /flow/{{ env.setup_sh }}
 {% endif %}
 
-RUN apt-get update && apt-get install -y runit
-
 EXPOSE 8080
 
 COPY ./connections/* /connections/