diff --git a/main_custom.py b/main_custom.py
index caec90390966fa3c7ad40161329eed98ab2a3aeb..1c9d5a5b187f3dff0fd11d4a61039ab9e6b67969 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -223,7 +223,7 @@ def main(args):
         data_train, data_val, data_test = common_dataset.load_data(path_train=args.dataset_train,
                                                                    path_val=args.dataset_val,
                                                                    path_test=args.dataset_test,
-                                                                   batch_size=args.batch_size, length=args.seq_length, pad = False, convert=True, vocab='unmod')
+                                                                   batch_size=args.batch_size, length=args.seq_length, pad = True, convert=True, vocab='unmod')
     elif args.forward == 'rt':
         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=args.seq_length)