diff --git a/docs/Project.toml b/docs/Project.toml
index b04849fe..0c856dad 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -27,7 +27,7 @@ Lux = "1"
 LuxCUDA = "0.3"
 MLDatasets = "0.7"
 MLUtils = "0.4"
-NonlinearSolve = "3"
+NonlinearSolve = "3, 4"
 OneHotArrays = "0.2"
 Optimisers = "0.3"
 OrdinaryDiffEq = "6"