1 files changed, 2 insertions, 2 deletions
diff --git a/sim.py b/sim.py
index 53f7f9a..7ba6472 100755
--- a/sim.py
+++ b/sim.py
@@ -107,7 +107,7 @@ class Simulation(object):
             self.gpu_pdf.clear_pdf()
 
         if nreps > 1:
-            iterable = repeating_iterator(iterable)
+            iterable = repeating_iterator(iterable, nreps)
 
         for ev in iterable:
             gpu_photons = gpu.GPUPhotons(ev.photons_beg)
@@ -138,7 +138,7 @@ class Simulation(object):
             iterable = self.photon_generator.generate_events(iterable)
 
         if nreps > 1:
-            iterable = repeating_iterator(iterable)
+            iterable = repeating_iterator(iterable, nreps)
 
         for ev in iterable:
             gpu_photons = gpu.GPUPhotons(ev.photons_beg)