From 90a8dd6ddfbc3faad04e892d46e5c23096222fa8 Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@gmail.com>
Date: Sat, 22 Mar 2025 14:22:46 +0530
Subject: [PATCH] fix: fix jump system hack

---
 src/systems/jumps/jumpsystem.jl | 25 +++++++------------------
 1 file changed, 7 insertions(+), 18 deletions(-)

diff --git a/src/systems/jumps/jumpsystem.jl b/src/systems/jumps/jumpsystem.jl
index 60637ed6d1..1ad6d30f59 100644
--- a/src/systems/jumps/jumpsystem.jl
+++ b/src/systems/jumps/jumpsystem.jl
@@ -4,25 +4,14 @@ const JumpType = Union{VariableRateJump, ConstantRateJump, MassActionJump}
 # call reset_aggregated_jumps!(integrator).
 # assumes iip
 function _reset_aggregator!(expr, integrator)
-    if expr isa Symbol
-        error("Error, encountered a symbol. This should not happen.")
+    @assert Meta.isexpr(expr, :function)
+    body = expr.args[end]
+    body = quote
+        $body
+        $reset_aggregated_jumps!($integrator)
     end
-    if expr isa LineNumberNode
-        return
-    end
-
-    if (expr.head == :function)
-        _reset_aggregator!(expr.args[end], integrator)
-    else
-        if expr.args[end] == :nothing
-            expr.args[end] = :(reset_aggregated_jumps!($integrator))
-            push!(expr.args, :nothing)
-        else
-            _reset_aggregator!(expr.args[end], integrator)
-        end
-    end
-
-    nothing
+    expr.args[end] = body
+    return nothing
 end
 
 """