diff --git a/Project.toml b/Project.toml
index cc23f0551..2411ae9be 100644
--- a/Project.toml
+++ b/Project.toml
@@ -15,7 +15,7 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 BSON = "0.3.2"
 Flux = "0.12"
 Functors = "0.2"
-NNlib = "0.7.34"
+NNlib = "0.7.34, 0.8"
 julia = "1.4"
 
 [extras]