diff --git a/docs/Project.toml b/docs/Project.toml
index 1692260..18de1ab 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -18,7 +18,7 @@ Lux = "1"
 ModelingToolkit = "9.9"
 ModelingToolkitNeuralNets = "1"
 ModelingToolkitStandardLibrary = "2.7"
-Optimization = "3.24"
+Optimization = "3.24, 4.0"
 OptimizationOptimisers = "0.2.1, 0.3"
 OrdinaryDiffEq = "6.74"
 Plots = "1"