diff --git a/lab8/Dockerfile b/lab8/Dockerfile
index f96010a..d538579 100644
--- a/lab8/Dockerfile
+++ b/lab8/Dockerfile
@@ -14,6 +14,7 @@ RUN pip3 install sacred
 RUN pip3 install pymongo
 RUN pip3 install GitPython
 RUN pip3 install mlflow
+RUN pip3 install mlflow.keras
 
 COPY ./lego_sets.csv ./
 COPY ./process_dataset.py ./