diff --git a/fidimag/common/driver_base.py b/fidimag/common/driver_base.py
index baf0c86b..0515449f 100644
--- a/fidimag/common/driver_base.py
+++ b/fidimag/common/driver_base.py
@@ -89,7 +89,7 @@ def set_integrator(self, integrator, use_jac):
         elif integrator == "sundials":
             self.integrator = CvodeSolver(self.spin, self.sundials_rhs)
         elif integrator == "euler" or integrator == "rk4":
-            self.integrator = StepIntegrator(self.spin, self.step_rhs)
+            self.integrator = StepIntegrator(self.spin, self.step_rhs, step=integrator)
         elif integrator == "scipy":
             self.integrator = ScipyIntegrator(self.spin, self.step_rhs)