diff --git a/Project.toml b/Project.toml
index 9a0b3b9b..4abdefb8 100644
--- a/Project.toml
+++ b/Project.toml
@@ -19,7 +19,7 @@ CUDA = "1, 2, 3, 4"
 ChainRulesCore = "0.8, 0.9, 0.10, 1"
 Flux = "0.11, 0.12, 0.13"
 JOLI = "0.7, 0.8"
-NNlib = "0.7, 0.8"
+NNlib = "0.7, 0.8, 0.9"
 TimerOutputs = "0.5"
 Wavelets = "0.9"
 julia = "1"