diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index d995920521420b57388b57e50a2af6710f02cbb3..a98228468d154cd9c9e8b71e9f30e3e7ec787beb 100644
--- a/modules/GSOC/E2_ESRGAN/lib/dataset.py
+++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py
@@ -155,7 +155,7 @@ class OpdNpyDataset:
         # dataset = dataset.cache()
         # dataset = dataset.prefetch(buffer_size=1)
 
-        file_idxs = np.arange(len(self.num_files))
+        file_idxs = np.arange(self.num_files)
         dataset = tf.data.Dataset.from_tensor_slices(list(file_idxs))
         dataset = dataset.shuffle(2000, reshuffle_each_iteration=True)
         dataset = dataset.batch(batch_size)