diff --git a/docker/Dockerfile b/docker/Dockerfile
index b763985c..4a8b86d5 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,8 +1,8 @@
 FROM python:3.10 as base
 
 WORKDIR /app
-ADD requirements.txt .
-RUN pip install -r requirements.txt && rm requirements.txt
+ADD pyproject.toml .
+RUN pip install . && rm pyproject.toml
 ENV PYTHONPATH=/app
 ADD pr_agent pr_agent
 
diff --git a/docker/Dockerfile.lambda b/docker/Dockerfile.lambda
index 404c8529..59e78a54 100644
--- a/docker/Dockerfile.lambda
+++ b/docker/Dockerfile.lambda
@@ -4,9 +4,9 @@ RUN yum update -y && \
     yum install -y gcc python3-devel && \
     yum clean all
 
-ADD requirements.txt .
-RUN pip install -r requirements.txt && rm requirements.txt
-RUN pip install mangum==16.0.0
+ADD pyproject.toml .
+RUN pip install . && rm pyproject.toml
+RUN pip install mangum==0.17.0
 COPY pr_agent/ ${LAMBDA_TASK_ROOT}/pr_agent/
 
 CMD ["pr_agent.servers.serverless.serverless"]