diff --git a/Project.toml b/Project.toml
index 7aec91b..90d8480 100644
--- a/Project.toml
+++ b/Project.toml
@@ -59,7 +59,7 @@ JLD2 = "0.1, 0.2, 0.3, 0.4"
 JSON = "0.21"
 KernelAbstractions = "0.6, 0.7, 0.9"
 NCDatasets = "0.11.7, 0.12.0"
-NNlib = "0.8"
+NNlib = "0.8, 0.9"
 NetCDF = "0.10, 0.11"
 Parameters = "0.12"
 Polynomials = "2, 3"