diff --git a/Project.toml b/Project.toml
index 0cd2931..a381fe1 100644
--- a/Project.toml
+++ b/Project.toml
@@ -19,6 +19,6 @@ Umlaut = "92992a2b-8ce5-4a9c-bb9d-58be9a7dc841"
 ChainRules = "1.43"
 ChainRulesCore = "1.15"
 FiniteDifferences = "0.12"
-NNlib = "0.8"
+NNlib = "0.8, 0.9"
 Umlaut = "0.5.1"
 julia = "1.6"