diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py
index 5f2146eac0b22ce8503aedde23604dd3cf89bb68..cd3881a7e3f907acc7e54eadbd8be89bb1d4c1f5 100644
--- a/modules/deeplearning/srcnn.py
+++ b/modules/deeplearning/srcnn.py
@@ -194,13 +194,13 @@ class SRCNN:
 
     def get_in_mem_data_batch(self, idxs, is_training):
         if is_training:
-            label_files = self.train_data_files
+            files = self.train_data_files
         else:
-            label_files = self.test_data_files
+            files = self.test_data_files
 
         label_s = []
         for k in idxs:
-            f = label_files[k]
+            f = files[k]
             nda = np.load(f)
             label_s.append(nda)