Bugfix for calc_W!

SciML · Jul 30, 2018 · 2713004 · 2713004
1 parent 696b119
commit 2713004
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/derivative_utils.jl b/src/derivative_utils.jl
@@ -282,7 +282,7 @@ function calc_W!(integrator, cache::OrdinaryDiffEqConstantCache, dtgamma, repeat
   if !W_transform
     if DiffEqBase.has_jac(f)
       J = f.jac(uprev, p, t)
-      if !isa(f, DiffEqBase.AbstractDiffEqLinearOperator)
+      if !isa(J, DiffEqBase.AbstractDiffEqLinearOperator)
         J = DiffEqArrayOperator(J)
       end
       W = WOperator(mass_matrix, dtgamma, J; transform=false)
@@ -297,7 +297,7 @@ function calc_W!(integrator, cache::OrdinaryDiffEqConstantCache, dtgamma, repeat
   else
     if DiffEqBase.has_jac(f)
       J = f.jac(uprev, p, t)
-      if !isa(f, DiffEqBase.AbstractDiffEqLinearOperator)
+      if !isa(J, DiffEqBase.AbstractDiffEqLinearOperator)
         J = DiffEqArrayOperator(J)
       end
       W = WOperator(mass_matrix, dtgamma, J; transform=true)