diff --git a/test/Project.toml b/test/Project.toml
index 25db89ce..c0cf1644 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -19,5 +19,5 @@ ForwardDiff = "0.10.12"
 NNlib = "0.7"
 ReverseDiff = "1.4.2"
 Tracker = "0.2.11"
-Zygote = "0.5.4"
+Zygote = "0.5.4, 0.6"
 julia = "1.3"