diff --git a/api/Dockerfile b/api/Dockerfile
index 3873964..0ecf4de 100644
--- a/api/Dockerfile
+++ b/api/Dockerfile
@@ -16,8 +16,9 @@ RUN poetry config virtualenvs.create false && \
     poetry install --no-root --no-cache --with api
 
 COPY api ta1-service/api
+COPY lib ta1-service/lib
 WORKDIR /ta1-service
 
 EXPOSE 8000
-ENV PYTHONPATH=/ta1-service/api
-CMD [ "uvicorn", "server:app", "--reload", "--host", "0.0.0.0", "--port", "8000" ]
+ENV PYTHONPATH=/ta1-service
+CMD [ "uvicorn", "api.server:app", "--reload", "--host", "0.0.0.0", "--port", "8000" ]
diff --git a/worker/Dockerfile b/worker/Dockerfile
index 562425b..2f45a14 100644
--- a/worker/Dockerfile
+++ b/worker/Dockerfile
@@ -19,5 +19,5 @@ ENV REDIS_HOST redis
 ENV REDIS_PORT 6379
 
 COPY worker worker
-WORKDIR /worker
+COPY lib lib
 CMD rq worker --url redis://$REDIS_HOST:$REDIS_PORT high default low
\ No newline at end of file