Update ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl

Co-authored-by: Lukas Devos <ldevos98@gmail.com>

Author: kshyatt

ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl

@@ -113,13 +113,8 @@ for (f, pb) in (
                 end
             end
             if !isa(arg, Const)
-                if !A_is_arg
-                    make_zero!(arg.dval)
-                elseif A_is_arg1
-                    make_zero!(arg.dval[2])
-                elseif A_is_arg2
-                    make_zero!(arg.dval[1])
-                end
+                A.dval === arg.dval[1] || make_zero!(arg.dval[1])
+                A.dval === arg.dval[2] || make_zero!(arg.dval[2])
             end
             return (nothing, nothing, nothing)
         end