diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index fd525c97456fe6f33769e2d9f7f21488e084d42c..e845ebf656379ab26eef24fbf5c97824dd3555b6 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -645,8 +645,8 @@ class SRCNN:
         return pred
 
     def run(self, directory, ckpt_dir=None, num_data_samples=50000):
-        train_data_files = glob.glob(directory+'data_train*.npy')
-        valid_data_files = glob.glob(directory+'data_valid*.npy')
+        train_data_files = glob.glob(directory+'data_train_*.npy')
+        valid_data_files = glob.glob(directory+'data_valid_*.npy')
 
         self.setup_pipeline(train_data_files, valid_data_files, num_data_samples)
         self.build_model()