From 90130b01cc6642efcfe496db1b4a0c20ee24df32 Mon Sep 17 00:00:00 2001
From: "William S. Moses" <gh@wsmoses.com>
Date: Sat, 14 Dec 2024 16:58:14 -0500
Subject: [PATCH] fixup

---
 src/Enzyme.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Enzyme.jl b/src/Enzyme.jl
index 4328a561e3..e2244171a5 100644
--- a/src/Enzyme.jl
+++ b/src/Enzyme.jl
@@ -370,7 +370,7 @@ Enzyme.autodiff(ReverseWithPrimal, x->x*x, Active(3.0))
     FTy = Core.Typeof(f.val)
 
     rt = if A isa UnionAll
-        Compiler.primal_return_type(mode, FTy, tt)
+        Compiler.primal_return_type(Reverse, FTy, tt)
     else
         eltype(A)
     end