From d56476c69c749a4b575b47d58c03cf366ab7134c Mon Sep 17 00:00:00 2001
From: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
Date: Fri, 4 Oct 2024 19:01:32 +0200
Subject: [PATCH] fix: use DI preparation result when initializing Jacobian
 (#472)

---
 src/internal/jacobian.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/internal/jacobian.jl b/src/internal/jacobian.jl
index 13feb8253..390c0947f 100644
--- a/src/internal/jacobian.jl
+++ b/src/internal/jacobian.jl
@@ -88,7 +88,7 @@ function JacobianCache(prob, alg, f::F, fu_, u, p; stats, autodiff = nothing,
                 if iip
                     DI.jacobian(f, fu, di_extras, autodiff, u, Constant(p))
                 else
-                    DI.jacobian(f, autodiff, u, Constant(p))
+                    DI.jacobian(f, di_extras, autodiff, u, Constant(p))
                 end
             end
         else