diff --git a/main_custom.py b/main_custom.py
index 78b8bb060d995465ba1fe325afd2a7b3c19bf0a5..7e6d34a8d93ff19e5db7cb03216ee113c7a4f1f3 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -202,7 +202,7 @@ def main(args):
                                                                    path_test=args.dataset_test,
                                                                    batch_size=args.batch_size, length=25, pad = True, convert=True, vocab='iapuc')
     elif args.forward == 'rt':
-        data_train, _, _ = dataloader.load_data(data_sources=[args.dataset_train,'database/data_holdout.csv','database/data_holdout.csv'],
+        data_train, _, _ = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_train,args.dataset_train],
                                                                batch_size=args.batch_size, length=25)
 
         _, data_val, data_test = common_dataset.load_data(path_train=args.dataset_val,