diff --git a/main_custom.py b/main_custom.py
index e34b7d1155adcdf2fe8da5cddbd7bbf162e98c25..fb99a74f1e5f9982a951d2ab70cfc8b334ca1a42 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -25,7 +25,7 @@ def train(model, data_train, epoch, optimizer, criterion_rt, criterion_intensity
     for param in model.parameters():
         param.requires_grad = True
     if forward == 'both':
-        print(data_train.dataset['Sequence'])
+        print(data_train.dataset.data['Sequence'])
         for seq, charge, rt, intensity in data_train:
             rt, intensity = rt.float(), intensity.float()
             if torch.cuda.is_available():