diff --git a/src/scimlfunctions.jl b/src/scimlfunctions.jl
index a503b0bf1..d81a63fc8 100644
--- a/src/scimlfunctions.jl
+++ b/src/scimlfunctions.jl
@@ -2509,7 +2509,8 @@ function ODEFunction{iip, specialize}(f;
             analytic, tgrad, jac,
             jvp, vjp, jac_prototype, sparsity, Wfact,
             Wfact_t, W_prototype, paramjac,
-            observed, _colorvec, sys, initializeprob, initializeprobmap, nlfunc)
+            observed, _colorvec, sys, initializeprob, update_initializeprob!,
+            initializeprobmap, initializeprobpmap, nlfunc)
     else
         ODEFunction{iip, specialize,
             typeof(_f), typeof(mass_matrix), typeof(analytic), typeof(tgrad),