diff --git a/Project.toml b/Project.toml
index dd645447..dd240427 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "ContinuousNormalizingFlows"
 uuid = "00b1973d-5b2e-40bf-8604-5c9c1d8f50ac"
-authors = ["Hossein Pourbozorg <prbzrg@gmail.com>"]
 version = "0.29.0"
+authors = ["Hossein Pourbozorg <prbzrg@gmail.com>"]
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
@@ -50,7 +50,7 @@ MLJModelInterface = "1"
 MLUtils = "0.4"
 NNlib = "0.9"
 Optimisers = "0.4"
-Optimization = "4"
+Optimization = "4, 5"
 OptimizationOptimisers = "0.3"
 OrdinaryDiffEqDefault = "1"
 Random = "1"