diff --git a/Dockerfile b/Dockerfile
index e0621e6fe335621d9c68c6ec1b150e4fd7a9d06c..dcde650ddc1df6117fa1ff30f0e5e073af55b4cb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -26,6 +26,7 @@ WORKDIR ${HOME}
 
 ADD wgan.py ${HOME}/wgan.py
 ADD wganHCAL.py ${HOME}/wganHCAL.py
+ADD wgan_ECAL_HCAL_2crit.py ${HOME}/wgan_ECAL_HCAL_2crit.py
 
 COPY ./models/* ${HOME}/models/
 COPY docker/krb5.conf /etc/krb5.conf