diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index cd7d453432fca7748726a18fe999020e69666cb0..fe562a660ac334cdb0b23ba7ff559e1cfba7784c 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -314,7 +314,7 @@ class SRCNN:
 
         trn_idxs = np.arange(len(train_data_files))
         np.random.shuffle(trn_idxs)
-        tst_idxs = np.arange(len(train_data_files))
+        tst_idxs = np.arange(len(test_data_files))
 
         self.get_train_dataset(trn_idxs)
         self.get_test_dataset(tst_idxs)