diff --git a/Project.toml b/Project.toml
index eeda494..660fa85 100644
--- a/Project.toml
+++ b/Project.toml
@@ -26,6 +26,8 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
+OrdinaryDiffEqCore = "bbf590c4-e513-4bbe-9b18-05decba2e5d8"
+
 
 [compat]
 DiffEqBase = "6"
diff --git a/src/CodeGeneration.jl b/src/CodeGeneration.jl
index 57ba5d9..df42b25 100644
--- a/src/CodeGeneration.jl
+++ b/src/CodeGeneration.jl
@@ -3,6 +3,7 @@
 
 using  OrderedCollections: OrderedDict, OrderedSet
 using  DataFrames
+using  OrdinaryDiffEqCore
 
 #=
 fieldnames(typeof(integrator)) = (:sol, :u, :du, :k, :t, :dt, :f, :p, :uprev, :uprev2, :duprev, :tprev, :alg,
@@ -1522,7 +1523,7 @@ function affectEvent!(integrator, stateEvent::Bool, eventIndex::Int)::Nothing
     end
 
     # Adapt step size
-    if eh.restart != NoRestart && supertype(typeof(integrator.alg)) == DifferentialEquations.OrdinaryDiffEq.OrdinaryDiffEqAdaptiveAlgorithm
+    if eh.restart != NoRestart && supertype(typeof(integrator.alg)) == OrdinaryDiffEqCore.OrdinaryDiffEqAdaptiveAlgorithm
         DifferentialEquations.auto_dt_reset!(integrator)
         DifferentialEquations.set_proposed_dt!(integrator, integrator.dt)
     end