diff --git a/adnc/data/utils/batch_generator.py b/adnc/data/utils/batch_generator.py
index 6f052b4..0b34763 100755
--- a/adnc/data/utils/batch_generator.py
+++ b/adnc/data/utils/batch_generator.py
@@ -62,7 +62,7 @@ class BatchGenerator():
                     self.order = self.data_set.rng.permutation(self.order)
 
     def __iter__(self):
-        return next(self)
+        return self
 
     def __next__(self):
         """