From 7f7ba11cb942f105530d6933c161befc26fcd24b Mon Sep 17 00:00:00 2001
From: Tor Erlend Fjelde <tor.erlend95@gmail.com>
Date: Fri, 26 Apr 2024 00:29:31 +0100
Subject: [PATCH] maybe now

---
 ext/AdvancedVIEnzymeExt.jl      | 2 +-
 ext/AdvancedVIReverseDiffExt.jl | 2 +-
 ext/AdvancedVIZygoteExt.jl      | 2 +-
 src/AdvancedVI.jl               | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/ext/AdvancedVIEnzymeExt.jl b/ext/AdvancedVIEnzymeExt.jl
index 86a8453e..025c3901 100644
--- a/ext/AdvancedVIEnzymeExt.jl
+++ b/ext/AdvancedVIEnzymeExt.jl
@@ -4,7 +4,7 @@ if isdefined(Base, :get_extension)
     using AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using Enzyme: Enzyme
 else
-    using ..AdvancedVI: ADTypes, AdvancedVI
+    using ..AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ..Enzyme: Enzyme
 end
 
diff --git a/ext/AdvancedVIReverseDiffExt.jl b/ext/AdvancedVIReverseDiffExt.jl
index 45d96368..dbe0d108 100644
--- a/ext/AdvancedVIReverseDiffExt.jl
+++ b/ext/AdvancedVIReverseDiffExt.jl
@@ -4,7 +4,7 @@ if isdefined(Base, :get_extension)
     using AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ReverseDiff: ReverseDiff
 else
-    using ..AdvancedVI: ADTypes, AdvancedVI
+    using ..AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ..ReverseDiff: ReverseDiff
 end
 
diff --git a/ext/AdvancedVIZygoteExt.jl b/ext/AdvancedVIZygoteExt.jl
index 4a28e00d..efb27732 100644
--- a/ext/AdvancedVIZygoteExt.jl
+++ b/ext/AdvancedVIZygoteExt.jl
@@ -4,7 +4,7 @@ if isdefined(Base, :get_extension)
     using AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using Zygote: Zygote
 else
-    using ..AdvancedVI: ADTypes, AdvancedVI
+    using ..AdvancedVI: AdvancedVI, ADTypes, DiffResults, Distributions
     using ..Zygote: Zygote
 end
 
diff --git a/src/AdvancedVI.jl b/src/AdvancedVI.jl
index f1dd6c84..59ae0e24 100644
--- a/src/AdvancedVI.jl
+++ b/src/AdvancedVI.jl
@@ -193,7 +193,7 @@ end
         @require ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267" include(
             "../ext/AdvancedVIReverseDiffExt.jl"
         )
-        @require ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444" include(
+        @require Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" include(
             "../ext/AdvancedVIZygoteExt.jl"
         )
         @require Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9" include(