diff --git a/Project.toml b/Project.toml
index 0b358f9..f8785fe 100644
--- a/Project.toml
+++ b/Project.toml
@@ -48,7 +48,7 @@ Sobol = "1"
 StatsBase = "0.34"
 UnicodePlots = "3"
 YAML = "0.4"
-Zygote = "0.6"
+Zygote = "0.6, 0.7"
 julia = "1.6"
 
 [extras]