diff --git a/Dockerfile b/Dockerfile
index 4022b36..75ff4a5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,6 +12,7 @@ RUN pip3 install numpy
 RUN pip3 install matplotlib
 RUN pip3 install torch
 RUN pip3 install sacred
+RUN pip3 install pymongo
 
 ARG CUTOFF
 ARG KAGGLE_USERNAME