whoops, missed a line

Author: MasonProtter

src/systems/diffeqs/sdesystem.jl

@@ -259,7 +259,7 @@ function generate_diffusion_function(sys::SDESystem, dvs = unknowns(sys),
         eqs = delay_to_function(sys, eqs)
     end
     if eqs isa AbstractMatrix && __num_isdiag_noise(eqs)
-        eqs = diag(eqs)
+        eqs = __get_num_diag_noise(eqs)
     end
     u = map(x -> time_varying_as_func(value(x), sys), dvs)
     p = if has_index_cache(sys) && get_index_cache(sys) !== nothing