diff --git a/wgan.py b/wgan.py
index 22c69b3d030bce98d6b95188f763d859884e668b..502f13f8e6e4b3e5fec179256f8ae4ae140b7d48 100644
--- a/wgan.py
+++ b/wgan.py
@@ -263,7 +263,7 @@ def run(args):
     print ("loading data")
     #dataset = HDF5Dataset('/eos/user/e/eneren/scratch/40GeV40k.hdf5', transform=None, train_size=40000)
     #dataset = HDF5Dataset('/eos/user/e/eneren/scratch/60GeV20k.hdf5', transform=None, train_size=20000)
-    dataset = HDF5Dataset('/eos/user/e/eneren/scratch/50GeV75k.hdf5', transform=None, train_size=75000)
+    dataset = HDF5Dataset('/eos/user/e/eneren/scratch/4060GeV.hdf5', transform=None, train_size=60000)
 
 
     sampler = DistributedSampler(dataset, shuffle=True)