diff --git a/main_custom.py b/main_custom.py
index cd1538d15b126efd78d8a7b127d476565334ea75..716029cfb65371b25803784cd9655a0fa40f3f35 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -216,7 +216,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,args.dataset_val,args.dataset_test],
+        data_train, data_val, data_test = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_val,args.dataset_test],
                                                                batch_size=args.batch_size, length=25)
 
     elif args.forward == 'transfer':