diff --git a/test/reduction.jl b/test/reduction.jl
index 391c5b07..d8faaa34 100644
--- a/test/reduction.jl
+++ b/test/reduction.jl
@@ -14,7 +14,7 @@ end
 prob = ODEProblem(f!, [0.5], (0.0, 1.0))
 
 function output_func(sol, ctx)
-    return last(sol), false
+    return sol.u[end], false
 end
 
 function prob_func(prob, ctx)