diff --git a/Dockerfile b/Dockerfile
index dc9cedf..aaf5fa9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,4 +12,4 @@ ARG KAGGLE_KEY
 RUN chmod a+x *.sh
 RUN ./download_dataset.sh
 
-CMD python power_plant_data_stats.py
+CMD python train_model.py