diff --git a/.docker-compose/Dockerfile b/.docker-compose/Dockerfile
index 62f8a78..b8715c1 100644
--- a/.docker-compose/Dockerfile
+++ b/.docker-compose/Dockerfile
@@ -6,6 +6,7 @@ RUN python -m pip install pulp
 RUN python -m pip install pydantic
 RUN python -m pip install daemonize
 RUN python -m pip install sqspy
+RUN python -m pip install -U pytest
 
 RUN mkdir /app
 WORKDIR /app
diff --git a/Dockerfile b/Dockerfile
index 58839d7..c41c12d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,6 +6,7 @@ RUN python -m pip install pulp
 RUN python -m pip install pydantic
 RUN python -m pip install daemonize
 RUN python -m pip install sqspy
+RUN python -m pip install -U pytest
 
 RUN mkdir /app
 WORKDIR /app