diff --git a/Dockerfile b/Dockerfile
index e44fa55..a972862 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,6 +11,7 @@ COPY ./nn_train.py ./
 
 RUN pip3 install pandas
 RUN pip3 install numpy
+RUN pip3 install matplotlib
 RUN pip3 install tensorflow
 RUN pip3 install keras
 RUN pip3 install sklearn