1 files changed, 1 insertions, 1 deletions
diff --git a/sim.py b/sim.py
index fce559e..c52a071 100755
--- a/sim.py
+++ b/sim.py
@@ -85,7 +85,7 @@ class Simulation(object):
 
     def propagate_photons(self, photons, max_steps=10):
         self.gpu_worker.load_photons(photons)
-        self.gpu_worker.propagate()
+        self.gpu_worker.propagate(max_steps=max_steps)
         return self.gpu_worker.get_photons()
 
 @profile_if_possible