Issues solving DifferentialEquations EnsembleProblem

Hello, I was trying to experiment with ensemble problems from differentialequations package. I’m very quickly running into a roadblock in that even the examples from documentation are producing errors. Below is an example directly from the documentation that errors on my system:

import DifferentialEquations as DE
prob = DE.ODEProblem((u, p, t) -> 1.01u, 0.5, (0.0, 1.0))
function prob_func(prob, i, repeat)
    DE.remake(prob, u0 = rand() * prob.u0)
end
ensemble_prob = DE.EnsembleProblem(prob, prob_func = prob_func)
sim = DE.solve(ensemble_prob, DE.Tsit5(), DE.EnsembleThreads(), trajectories = 10)
import Plots;
Plots.plot(sim);

The error:
no method matching prob_func(::SciMLBase.ODEProblem{…}, ::SciMLBase.EnsembleContext{…})
The function prob_func exists, but no method is defined for this combination of argument types.

Closest candidates are:
prob_func(::Any, ::Any, ::Any)
@ Main c:\Users\tedri\Documents\Scratch\test.jl:3

It would seem that the example arguments to prob_func are incorrect, but I am not sure how to proceed as this is direct from the documentation.

My packages:

(Scratch) pkg> status
Status C:\Users\tedri\Documents\Scratch\Project.toml
[0c46a032] DifferentialEquations v8.0.0
[91a5bcdd] Plots v1.41.6

Julia v1.12.6