diff --git a/src/festim/stepsize.py b/src/festim/stepsize.py
index ccafc827f..a94910f03 100644
--- a/src/festim/stepsize.py
+++ b/src/festim/stepsize.py
@@ -131,8 +131,7 @@ def modify_value(self, value, nb_iterations, t=None):
         else:
             updated_value = value
 
-        max_step = self.get_max_stepsize(t)
-        if max_step:
+        if max_step := self.get_max_stepsize(t):
             if updated_value > max_step:
                 updated_value = max_step