diff --git a/bench/Project.toml b/bench/Project.toml
index 0daa4b74..713b95f2 100644
--- a/bench/Project.toml
+++ b/bench/Project.toml
@@ -36,5 +36,5 @@ Random = "1"
 ReverseDiff = "1"
 SimpleUnPack = "1"
 StableRNGs = "1"
-Zygote = "0.6"
+Zygote = "0.6, 0.7"
 julia = "1.10, 1.11.2"
diff --git a/test/Project.toml b/test/Project.toml
index 3a71865b..927ac810 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -48,5 +48,5 @@ Statistics = "1"
 StatsBase = "0.34"
 Test = "1"
 Tracker = "0.2.20"
-Zygote = "0.6.63"
+Zygote = "0.6.63, 0.7"
 julia = "1.10, 1.11.2"