diff --git a/Dockerfile b/Dockerfile
index 67c4038..ababccf 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -20,7 +20,8 @@ ENV PATH="$POETRY_HOME/bin:$PATH"
 RUN curl -sSL https://install.python-poetry.org | python3 -
 
 COPY ./data ./data
-COPY pyproject.toml poetry.lock banner.txt config.yaml ./src ./image_prediction ./
+COPY ./image_prediction ./image_prediction
+COPY pyproject.toml poetry.lock banner.txt config.yaml ./src ./
 
 RUN poetry config virtualenvs.create false && \
     poetry config installer.max-workers 10 && \
diff --git a/Dockerfile_tests b/Dockerfile_tests
index 28247b7..561aca7 100644
--- a/Dockerfile_tests
+++ b/Dockerfile_tests
@@ -20,7 +20,8 @@ ENV PATH="$POETRY_HOME/bin:$PATH"
 RUN curl -sSL https://install.python-poetry.org | python3 -
 
 COPY ./data ./data
-COPY pyproject.toml poetry.lock banner.txt config.yaml ./src ./image_prediction ./
+COPY ./image_prediction ./image_prediction
+COPY pyproject.toml poetry.lock banner.txt config.yaml ./src ./
 
 RUN poetry config virtualenvs.create false && \
     poetry config installer.max-workers 10 && \