diff --git a/lib/NonlinearSolveBase/ext/NonlinearSolveBaseMooncakeExt.jl b/lib/NonlinearSolveBase/ext/NonlinearSolveBaseMooncakeExt.jl
index 91b901099..bed496361 100644
--- a/lib/NonlinearSolveBase/ext/NonlinearSolveBaseMooncakeExt.jl
+++ b/lib/NonlinearSolveBase/ext/NonlinearSolveBaseMooncakeExt.jl
@@ -9,7 +9,7 @@ import Mooncake: rrule!!, CoDual, zero_fcodual, @is_primitive,
 @from_rrule(MinimalCtx,
     Tuple{
         typeof(NonlinearSolveBase.solve_up),
-        SciMLBase.AbstractDEProblem,
+        SciMLBase.AbstractNonlinearProblem,
         Union{Nothing, SciMLBase.AbstractSensitivityAlgorithm},
         Any,
         Any,
@@ -21,7 +21,7 @@ import Mooncake: rrule!!, CoDual, zero_fcodual, @is_primitive,
 @from_rrule(MinimalCtx,
     Tuple{
         typeof(NonlinearSolveBase.solve_up),
-        SciMLBase.AbstractDEProblem,
+        SciMLBase.AbstractNonlinearProblem,
         Union{Nothing, SciMLBase.AbstractSensitivityAlgorithm},
         Any,
         Any