diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 6322e3c7c1dfa2e49acca2980fcab744cf5899ef..0bf3e8f3bda6a7774980fc6093459f23ba37ff46 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -21,4 +21,4 @@ check_python_run:
   image: pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime
   script: 
     - pip install pyflakes
-    - pyflakes $CI_PROJECT_DIR/wgan_ECAL_HCAL_2crit.py
+    - pyflakes $CI_PROJECT_DIR/wgan_ECAL_HCAL_3crit.py
diff --git a/Dockerfile b/Dockerfile
index dcde650ddc1df6117fa1ff30f0e5e073af55b4cb..f31165b449627cd83e471d7ad02b8a25accc2f9a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -26,7 +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
+ADD wgan_ECAL_HCAL_3crit.py ${HOME}/wgan_ECAL_HCAL_3crit.py
 
 COPY ./models/* ${HOME}/models/
 COPY docker/krb5.conf /etc/krb5.conf