..

dweindl · Feb 20, 2024 · 2f68856 · 2f68856
1 parent 2cf59c0
commit 2f68856
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/sdist/amici/de_export.py b/python/sdist/amici/de_export.py
@@ -1191,8 +1191,9 @@ def get_rate(symbol: sp.Symbol):
             topo_exprs = toposort_symbols(
                 dict(zip(self.sym("w"), self.eq("w")))
             )
+            old_syms = tuple(self._syms["w"])
             topo_expr_syms = tuple(topo_exprs.keys())
-            new_order = [topo_expr_syms.index(s) for s in self.sym("w")]
+            new_order = [old_syms.index(s) for s in topo_expr_syms]
             self._expressions = [self._expressions[i] for i in new_order]
             self._syms["w"] = sp.Matrix(topo_expr_syms)
             self._eqs["w"] = sp.Matrix(list(topo_exprs.values()))