diff --git a/wganHCAL.py b/wganHCAL.py
index 3696feb2627f6bf7357ada02de918cac6533c598..683a7568ef52edbbaf2ed84cd3c0d4f906ced667 100644
--- a/wganHCAL.py
+++ b/wganHCAL.py
@@ -274,7 +274,7 @@ def run(args):
 
 
     ## HCAL Generator and critic
-    mCrit = CriticEMB(args.ndf, emb_size=64).to(device)
+    mCrit = CriticEMB(args.ndf, size_embed=64).to(device)
     mGen = Hcal_ecalEMB(args.ngf, 32, args.nz, emb_size=32).to(device)
     
     ## ECAL GENERATOR