diff --git a/Project.toml b/Project.toml
index 64c69ab..0f228fc 100644
--- a/Project.toml
+++ b/Project.toml
@@ -56,7 +56,7 @@ StanSample = "6, 7"
 StructArrays = "0.7"
 TransformVariables = "0.8"
 TransformedLogDensities = "1"
-Turing = "0.38, 0.39"
+Turing = "0.38, 0.39, 0.40"
 julia = "1.10"
 
 [extras]
diff --git a/src/turing_inference.jl b/src/turing_inference.jl
index 9cea743..50d55a3 100644
--- a/src/turing_inference.jl
+++ b/src/turing_inference.jl
@@ -1,5 +1,5 @@
 function turing_inference(
-        prob::Union{DiffEqBase.DEProblem, DiffEqBase.AbstractNonlinearProblem},
+        prob::SciMLBase.AbstractSciMLProblem,
         alg,
         t,
         data,