Update ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl

Co-authored-by: Jutho <Jutho@users.noreply.github.com>

Author: kshyatt

ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl

@@ -324,9 +324,12 @@ for (f, trunc_f, full_f, pb) in (
         cache_DV = (DV.val !== ret) || EnzymeRules.overwritten(config)[3] ? copy.(ret) : nothing
         DV′, ind = truncate($trunc_f, ret, alg.val.trunc)
         primal = EnzymeRules.needs_primal(config) ? DV′ : nothing
-        dret = (Diagonal(zero(diagview(DV′[1]))), zero(DV′[2]))
-        shadow = EnzymeRules.needs_shadow(config) ? dret : nothing
-        return EnzymeRules.AugmentedReturn(primal, shadow, (cache_DV, dret, ind))
+        dret = if EnzymeRules.needs_shadow(config)
+            (Diagonal(zero(diagview(DV′[1]))), zero(DV′[2]))
+        else
+            nothing
+        end
+        return EnzymeRules.AugmentedReturn(primal, dret, (cache_DV, dret, ind))
     end
     @eval function EnzymeRules.reverse(
             config::EnzymeRules.RevConfigWidth{1},