diff --git a/src/solve.jl b/src/solve.jl
index 633e0246d..235f692d6 100644
--- a/src/solve.jl
+++ b/src/solve.jl
@@ -1093,9 +1093,13 @@ function solve(prob::AbstractDEProblem, args...; sensealg = nothing,
     p = p !== nothing ? p : prob.p
 
     if wrap isa Val{true}
-        wrap_sol(solve_up(prob, sensealg, u0, p, args...; kwargs...))
+        wrap_sol(solve_up(prob, sensealg, u0, p, args...; 
+        originator = set_mooncakeoriginator_if_mooncake(SciMLBase.ChainRulesOriginator()),
+        kwargs...))
     else
-        solve_up(prob, sensealg, u0, p, args...; kwargs...)
+        solve_up(prob, sensealg, u0, p, args...; 
+        originator = set_mooncakeoriginator_if_mooncake(SciMLBase.ChainRulesOriginator()),
+        kwargs...)
     end
 end