diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 9587044a84..039ab6a24e 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -46,7 +46,7 @@ jobs:
 
   build-publish-docker-images:
     runs-on: ubuntu-latest
-    needs: get-version
+    needs: [get-version, publish-python-sdk]
     strategy:
       matrix:
         component: [feature-server, feature-server-python-aws, feature-server-java, feature-transformation-server]
diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
index 990c1fd8f0..c95c515fb4 100644
--- a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
+++ b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
@@ -8,8 +8,8 @@ RUN apt update && \
         build-essential
 
 RUN pip install pip --upgrade
-COPY . .
-RUN pip install ".[aws,gcp,snowflake,redis,go,mysql,postgres]"
+RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]"
+
 
 RUN apt update
 RUN apt install -y -V ca-certificates lsb-release wget
diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
index 990c1fd8f0..ecbc199a5b 100644
--- a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
+++ b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
@@ -9,7 +9,8 @@ RUN apt update && \
 
 RUN pip install pip --upgrade
 COPY . .
-RUN pip install ".[aws,gcp,snowflake,redis,go,mysql,postgres]"
+
+RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]"
 
 RUN apt update
 RUN apt install -y -V ca-certificates lsb-release wget