whoops, missed a line

SciML · Jul 31, 2024 · 0a3f62c · 0a3f62c
1 parent bb403dd
commit 0a3f62c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/systems/diffeqs/sdesystem.jl b/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