diff --git a/.idea/LC-MS-RT-prediction.iml b/.idea/LC-MS-RT-prediction.iml
index 82f955b7bb3fb05d7308a08b9a3143de1c4bed5c..81c1f054c8f6682138a3e8066d2e19396c22b814 100644
--- a/.idea/LC-MS-RT-prediction.iml
+++ b/.idea/LC-MS-RT-prediction.iml
@@ -9,7 +9,7 @@
       <excludeFolder url="file://$MODULE_DIR$/.venv" />
       <excludeFolder url="file://$MODULE_DIR$/.venv2" />
     </content>
-    <orderEntry type="inheritedJdk" />
+    <orderEntry type="jdk" jdkName="Python 3.11 (LC-MS-RT-prediction)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>
\ No newline at end of file
diff --git a/main_custom.py b/main_custom.py
index ea218998d1c57ad39189898d949f0e52208349ec..1118c9aad5b7832c9bc62d7180de04ece7454e16 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 = False, convert=True, vocab='iapuc')
     elif args.forward == 'rt':
-        data_train, data_val, data_test = dataloader.load_data(data_sources=args.dataset_train,
+        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)
 
     print('\nData loaded')