From 939ddc035d9e5701d2aad9cfb26e06262cbf9986 Mon Sep 17 00:00:00 2001
From: "William S. Moses" <gh@wsmoses.com>
Date: Sat, 14 Dec 2024 17:05:06 -0500
Subject: [PATCH] fix

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

diff --git a/src/Enzyme.jl b/src/Enzyme.jl
index e2244171a5..ca133e0025 100644
--- a/src/Enzyme.jl
+++ b/src/Enzyme.jl
@@ -687,7 +687,7 @@ code, as well as high-order differentiation.
     A2 = A
 
     if A isa UnionAll
-        rt = Compiler.primal_return_type(mode, FTy, tt)
+        rt = Compiler.primal_return_type(Reverse, FTy, tt)
         A2 = A{rt}
         if rt == Union{}
             rt = Nothing