diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py
index e7ffa10f5811df0f6beb96bcbc1a232298d34773..7313d92339846f91c65826fe9b548baa597bf975 100644
--- a/modules/deeplearning/srcnn.py
+++ b/modules/deeplearning/srcnn.py
@@ -306,7 +306,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)