diff --git a/Project.toml b/Project.toml
index 0d81ed4..42dee94 100644
--- a/Project.toml
+++ b/Project.toml
@@ -28,5 +28,5 @@ NNlib = "0.9"
 SpecialFunctions = "2"
 SymbolicUtils = "3"
 Symbolics = "6"
-Zygote = "0.6"
+Zygote = "0.6, 0.7"
 julia = "1.10"