From ab15e8b9e1142b237ade7f20c1eb139b2dfe5e08 Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Sun, 19 Jan 2025 20:49:53 -0500
Subject: [PATCH] fix: call with reactant

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

diff --git a/src/Ops.jl b/src/Ops.jl
index e001ce9fc..ac1b1b61a 100644
--- a/src/Ops.jl
+++ b/src/Ops.jl
@@ -2118,7 +2118,7 @@ end
                     TracedRNumber{Reactant.unwrapped_eltype(arg)}, arg
                 )
             end
-            return f(scalar_args...)
+            return Reactant.call_with_reactant(f, scalar_args...)
         end
     end