diff --git a/infra/docker/jobservice/Dockerfile b/infra/docker/jobservice/Dockerfile
index bad0983f59..8fbf1a7159 100644
--- a/infra/docker/jobservice/Dockerfile
+++ b/infra/docker/jobservice/Dockerfile
@@ -10,6 +10,9 @@ COPY protos protos
 # Install necessary tools for later steps
 RUN apt-get update && apt-get -y install make git wget
 
+# Install CI requirements (only needed for running tests)
+RUN pip install -r sdk/python/requirements-ci.txt
+
 # Install Python dependencies
 RUN make compile-protos-python
 
diff --git a/infra/docker/jupyter/Dockerfile b/infra/docker/jupyter/Dockerfile
index 5126aaffec..000855806d 100644
--- a/infra/docker/jupyter/Dockerfile
+++ b/infra/docker/jupyter/Dockerfile
@@ -8,12 +8,12 @@ COPY Makefile Makefile
 COPY protos protos
 COPY tests tests
 
-# Install Python dependencies
-RUN make compile-protos-python
-
 # Install CI requirements (only needed for running tests)
 RUN pip install -r sdk/python/requirements-ci.txt
 
+# Install Python dependencies
+RUN make compile-protos-python
+
 # Install Feast SDK
 RUN git init .
 COPY README.md README.md