diff --git a/bayesflow/utils/integrate.py b/bayesflow/utils/integrate.py
index 97c1b5729..ab652f401 100644
--- a/bayesflow/utils/integrate.py
+++ b/bayesflow/utils/integrate.py
@@ -199,7 +199,7 @@ def body(_state, _time, _step_size, _step):
         time_remaining = stop_time - (_time + _step_size)
 
         min_step_size = time_remaining / (max_steps - _step)
-        max_step_size = time_remaining / keras.ops.maximum(min_steps - _step, 1)
+        max_step_size = time_remaining / keras.ops.maximum(min_steps - _step, 1.0)
 
         # reorder
         min_step_size, max_step_size = (