diff --git a/build_automation/cloudberry/scripts/configure-cloudberry.sh b/build_automation/cloudberry/scripts/configure-cloudberry.sh
index 587ba10..a1a6985 100755
--- a/build_automation/cloudberry/scripts/configure-cloudberry.sh
+++ b/build_automation/cloudberry/scripts/configure-cloudberry.sh
@@ -145,7 +145,6 @@ execute_cmd ./configure --prefix=/usr/local/cloudberry-db \
             --with-perl \
             --with-pgport=5432 \
             --with-python \
-            --with-pythonsrc-ext \
             --with-ssl=openssl \
             --with-openssl \
             --with-uuid=e2fs \
diff --git a/images/docker/cbdb/build/rocky8/Dockerfile b/images/docker/cbdb/build/rocky8/Dockerfile
index cf37389..468c4d0 100644
--- a/images/docker/cbdb/build/rocky8/Dockerfile
+++ b/images/docker/cbdb/build/rocky8/Dockerfile
@@ -119,6 +119,9 @@ RUN dnf makecache && \
         procps-ng \
         python3 \
         python3-devel \
+        python3-psutil \
+        python3-psycopg2 \
+        python3-pyyaml \
         readline-devel \
         rsync \
         sshpass \
diff --git a/images/docker/cbdb/build/rocky9/Dockerfile b/images/docker/cbdb/build/rocky9/Dockerfile
index fa108b1..96070a5 100644
--- a/images/docker/cbdb/build/rocky9/Dockerfile
+++ b/images/docker/cbdb/build/rocky9/Dockerfile
@@ -130,6 +130,9 @@ RUN dnf makecache && \
         perl-Test-Simple \
         perl-core \
         python3-devel \
+        python3-psutil \
+        python3-psycopg2 \
+        python3-pyyaml \
         python3-pytest \
         readline-devel \
         zlib-devel && \