diff --git a/src/cache.jl b/src/cache.jl
index 03bedcb..6e7778e 100644
--- a/src/cache.jl
+++ b/src/cache.jl
@@ -22,7 +22,7 @@ function OptimizationCache(prob::SciMLBase.OptimizationProblem, opt, data;
         reltol::Union{Number, Nothing} = nothing,
         progress = false,
         kwargs...)
-    reinit_cache = Optimization.ReInitCache(prob.u0, prob.p)
+    reinit_cache = OptimizationBase.ReInitCache(prob.u0, prob.p)
     num_cons = prob.ucons === nothing ? 0 : length(prob.ucons)
     f = Optimization.instantiate_function(prob.f, reinit_cache, prob.f.adtype, num_cons)
     return OptimizationCache(f, reinit_cache, prob.lb, prob.ub, prob.lcons,