diff --git a/maxdiffusion_dependencies.Dockerfile b/maxdiffusion_dependencies.Dockerfile
index 5986b72..778e9e4 100644
--- a/maxdiffusion_dependencies.Dockerfile
+++ b/maxdiffusion_dependencies.Dockerfile
@@ -45,7 +45,7 @@ ARG JAX_VERSION
 ENV ENV_JAX_VERSION=$JAX_VERSION
 
 # Set the working directory in the container
-WORKDIR /deps
+WORKDIR /app
 
 # Copy all files from local workspace into docker container
 COPY . .
@@ -54,8 +54,5 @@ RUN echo "Running command: bash setup.sh MODE=$ENV_MODE JAX_VERSION=$ENV_JAX_VER
 
 RUN --mount=type=cache,target=/root/.cache/pip bash setup.sh MODE=${ENV_MODE} JAX_VERSION=${ENV_JAX_VERSION}
 
-# Install MaxDiffusion
-RUN pip install .
-
 # Cleanup
 RUN rm -rf /root/.cache/pip
\ No newline at end of file
diff --git a/maxdiffusion_jax_stable_stack_tpu.Dockerfile b/maxdiffusion_jax_stable_stack_tpu.Dockerfile
index 188640c..ff883b9 100644
--- a/maxdiffusion_jax_stable_stack_tpu.Dockerfile
+++ b/maxdiffusion_jax_stable_stack_tpu.Dockerfile
@@ -7,19 +7,16 @@ ARG COMMIT_HASH
 
 ENV COMMIT_HASH=$COMMIT_HASH
 
-RUN mkdir -p /deps
+RUN mkdir -p /app
 
 # Set the working directory in the container
-WORKDIR /deps
+WORKDIR /app
 
 # Copy all files from local workspace into docker container
 COPY . .
 
 # Install Maxdiffusion jax stable stack requirements
-RUN pip install -r /deps/requirements_with_jax_stable_stack.txt
-
-# Install MaxDiffusion
-RUN pip install .
+RUN pip install -r /app/requirements_with_jax_stable_stack.txt
 
 # Run the script available in JAX-Stable-Stack base image to generate the manifest file
 RUN bash /jax-stable-stack/generate_manifest.sh PREFIX=maxdiffusion COMMIT_HASH=$COMMIT_HASH
\ No newline at end of file
diff --git a/maxdiffusion_runner.Dockerfile b/maxdiffusion_runner.Dockerfile
index 9c83037..50a93e3 100644
--- a/maxdiffusion_runner.Dockerfile
+++ b/maxdiffusion_runner.Dockerfile
@@ -1,12 +1,10 @@
 ARG BASEIMAGE=maxdiffusion_base_image
 FROM $BASEIMAGE
 
-#FROM maxdiffusion_base_image
-
 # Set the working directory in the container
-WORKDIR /deps
+WORKDIR /app
 
 # Copy all files from local workspace into docker container
 COPY . .
 
-WORKDIR /deps
\ No newline at end of file
+WORKDIR /app
\ No newline at end of file