Fix for MTK change

SciML · May 26, 2020 · b2ce50f · b2ce50f
1 parent e33fdde
commit b2ce50f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/stan_inference.jl b/src/stan_inference.jl
@@ -58,7 +58,7 @@ function stan_inference(prob::DiffEqBase.DEProblem,t,data,priors = nothing,
   save_idxs = something(save_idxs, 1:length_of_y)
   length_of_params = length(vars)
   if isnothing(diffeq_string)
-    sys = first(ModelingToolkit.modelingtoolkitize(prob))
+    sys = ModelingToolkit.modelingtoolkitize(prob)
     length_of_parameter = length(sys.ps) + sample_u0 * length(save_idxs)
   else
     length_of_parameter = length(prob.p) + sample_u0 * length(save_idxs)