From 443c731a4e0c0707215bfc6aa12db8bac56f45fe Mon Sep 17 00:00:00 2001
From: Hong Ge <3279477+yebai@users.noreply.github.com>
Date: Wed, 21 Aug 2024 20:43:10 +0100
Subject: [PATCH] Update test/ad/distributions.jl

---
 test/ad/distributions.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/ad/distributions.jl b/test/ad/distributions.jl
index 66c910b..699bc7a 100644
--- a/test/ad/distributions.jl
+++ b/test/ad/distributions.jl
@@ -62,8 +62,8 @@
         DistSpec(Poisson, (0.5,), [1, 1]),
 
         # Enzyme: no forward rule for ccall 
-        DistSpec(Skellam, (1.0, 2.0), -2, broken=(:Zygote,)),
-        DistSpec(Skellam, (1.0, 2.0), [-2, -2], broken=(:Zygote,)),
+        DistSpec(Skellam, (1.0, 2.0), -2, broken=(:Enzyme,)),
+        DistSpec(Skellam, (1.0, 2.0), [-2, -2], broken=(:Enzyme,)),
 
         DistSpec(PoissonBinomial, ([0.5, 0.5],), 0),