diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py
index 764df959cccbef554f61ba3c9e2b9254f72db1e4..d3084cf334e0ea32a5dda4741de64a58ce538bd7 100644
--- a/modules/deeplearning/unet_l1b_l2.py
+++ b/modules/deeplearning/unet_l1b_l2.py
@@ -214,7 +214,6 @@ class UNET:
         #         print(e)
 
     def get_in_mem_data_batch(self, idxs, is_training):
-        np.sort(idxs)
 
         dat_files = []
         lbl_files = []
@@ -236,11 +235,11 @@ class UNET:
             label = np.expand_dims(label, axis=3)
         else:
             for k in idxs:
-                f = self.train_data_files[k]
+                f = self.test_data_files[k]
                 nda = np.load(f)
                 dat_files.append(nda)
 
-                f = self.train_label_files[k]
+                f = self.test_label_files[k]
                 nda = np.load(f)
                 lbl_files.append(nda)