diff --git a/Project.toml b/Project.toml
index b667f554..da764631 100644
--- a/Project.toml
+++ b/Project.toml
@@ -61,7 +61,7 @@ Memoize = "0.4"
 ModelingToolkit = "8, 9"
 NNlib = "0.8, 0.9"
 Optimisers = "0.2, 0.3"
-Optimization = "3"
+Optimization = "3, 4"
 OptimizationOptimisers = "0.1, 0.2"
 PackageExtensionCompat = "1"
 ProgressBars = "1.5"