up

SciML · Nov 14, 2023 · cf95d0c · cf95d0c
1 parent f210093
commit cf95d0c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/registered_functions.jl b/src/registered_functions.jl
@@ -142,5 +142,6 @@ function expand_registered_functions(eq::Equation)
 end
 # If applied to a ReactionSystem, applied function to all Reactions and other Equations, and return updated system.
 function expand_registered_functions(rs::ReactionSystem)
-    @set rs.eqs = [Catalyst.expand_registered_functions(eq) for eq in rs.eqs]
+    rs = @set rs.eqs = [Catalyst.expand_registered_functions(eq) for eq in rs.eqs]
+    return @set rs.rxs = [Catalyst.expand_registered_functions(rx) for rx in rs.rxs]
 end