..

python/sdist/amici/de_export.py

@@ -2760,11 +2760,9 @@ def _process_heavisides(
             # `tmp_new` is None if the root is not time-dependent.
             if tmp_new is None:
                 continue
-
-            heavisides.append((sp.Heaviside(tmp_old), tmp_new))
-
             # For Heavisides, we need to add the negative function as well
             self._get_unique_root(sp.sympify(-tmp_old), roots)
+            heavisides.append((sp.Heaviside(tmp_old), tmp_new))
 
         if heavisides:
             # only apply subs if necessary