diff --git a/Project.toml b/Project.toml
index 617dfe1..dbee83b 100644
--- a/Project.toml
+++ b/Project.toml
@@ -18,7 +18,7 @@ ParameterizedQuantumControlOptimizationExt = "Optimization"
 
 [compat]
 LinearAlgebra = "1"
-Optimization = "3"
+Optimization = "3, 4"
 QuantumControl = ">=0.11.0"
 QuantumGradientGenerators = ">=0.1.8"
 julia = "1.9"