diff --git a/src/solve/mirk.jl b/src/solve/mirk.jl index 342df31c..79be3d69 100644 --- a/src/solve/mirk.jl +++ b/src/solve/mirk.jl @@ -170,6 +170,6 @@ function SciMLBase.solve!(cache::MIRKCache) end end - return DiffEqBase.build_solution(prob, alg, mesh, + return DiffEqBase.build_solution(prob, alg, cache.mesh, [reshape(y, cache.in_size) for y in cache.y₀]; retcode = info) end