diff --git a/prosit_data_merge.py b/prosit_data_merge.py
index 32f88ad0534b6d3ed0ba7efa67b5828c6976d96a..1404420178aa74b6ce73b4780f2075e7727696d5 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/intensity_train.npy',
+             'data/intensity/precursor_charge_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/intensity_holdout.npy',
+             'data/intensity/precursor_charge_holdout.npy',
              'data/intensity/precursor_charge_holdout.npy')