diff --git a/src/sensitivity_algorithms.jl b/src/sensitivity_algorithms.jl index 3746c5e26..2073522f5 100644 --- a/src/sensitivity_algorithms.jl +++ b/src/sensitivity_algorithms.jl @@ -993,7 +993,8 @@ Base.@pure function SteadyStateAdjoint(; chunk_size = 0, autodiff = true, SteadyStateAdjoint{chunk_size, autodiff, diff_type, typeof(autojacvec), typeof(linsolve) }(autojacvec, linsolve) end -function setvjp(sensealg::SteadyStateAdjoint{CS, AD, FDT, LS}, vjp) where {CS, AD, FDT, LS} +function setvjp(sensealg::SteadyStateAdjoint{CS, AD, FDT, VJP, LS}, + vjp) where {CS, AD, FDT, VJP, LS} SteadyStateAdjoint{CS, AD, FDT, typeof(vjp), LS}(vjp, sensealg.linsolve) end