diff --git a/prosit_data_merge.py b/prosit_data_merge.py
index 8cc8125091127df85a6eded82d254d9124880541..32f88ad0534b6d3ed0ba7efa67b5828c6976d96a 100644
--- a/prosit_data_merge.py
+++ b/prosit_data_merge.py
@@ -70,7 +70,7 @@ def alphabetical_to_numerical(seq):
 
 sources = ('data/intensity/sequence_train.npy',
              'data/intensity/intensity_train.npy',
-             'data/intensity/collision_energy_train.npy',
+             'data/intensity/intensity_train.npy',
              'data/intensity/precursor_charge_train.npy')
 
 
@@ -110,7 +110,7 @@ np.save('data/intensity/irt_train.npy',data_int.irt)
 
 sources = ('data/intensity/sequence_holdout.npy',
              'data/intensity/intensity_holdout.npy',
-             'data/intensity/collision_energy_holdout.npy',
+             'data/intensity/intensity_holdout.npy',
              'data/intensity/precursor_charge_holdout.npy')