diff --git a/Dockerfile b/Dockerfile
index 55bd231..1af2b3b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -48,7 +48,8 @@ RUN micromamba create -n $ENV_NAME -y && \
     neptune==1.10.2 -y && \
     /usr/local/bin/micromamba run -n $ENV_NAME pip3 install opencv-python==4.9.0.80 \
     tensorflow-addons==0.23.0 \
-    cupy-cuda12x==13.3.0 && \
+    cupy-cuda12x==13.3.0 \
+    pytest && \
     /usr/local/bin/micromamba run -n $ENV_NAME python -m pip install .
 
 # Default command when the container starts (optional)