From a5030a2d10e202dc7649b9ecdc69eff834a6db24 Mon Sep 17 00:00:00 2001
From: Nathanael Bosch <nathanael.bosch@uni-tuebingen.de>
Date: Fri, 30 Aug 2024 11:14:04 +0200
Subject: [PATCH] Resolve JET.jl issue

---
 src/caches.jl | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/caches.jl b/src/caches.jl
index 20317077c..0253c5063 100644
--- a/src/caches.jl
+++ b/src/caches.jl
@@ -251,7 +251,9 @@ function OrdinaryDiffEq.alg_cache(
     )
 end
 
-get_uf(f, t, p, ::Val{true}) = OrdinaryDiffEq.UJacobianWrapper(f, t, p)
-get_uf(f, t, p, ::Val{false}) = OrdinaryDiffEq.UDerivativeWrapper(f, t, p)
+get_uf(f, t, p, ::Val{true}) =
+    OrdinaryDiffEq.OrdinaryDiffEqDifferentiation.UJacobianWrapper(f, t, p)
+get_uf(f, t, p, ::Val{false}) =
+    OrdinaryDiffEq.OrdinaryDiffEqDifferentiation.UDerivativeWrapper(f, t, p)
 
 OrdinaryDiffEq.get_fsalfirstlast(cache::EKCache, rate_prototype) = nothing, nothing