1 files changed, 2 insertions, 2 deletions
diff --git a/sim.py b/sim.py
index 931858f..26835f9 100755
--- a/sim.py
+++ b/sim.py
@@ -110,7 +110,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)
@@ -139,7 +139,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)