diff --git a/Project.toml b/Project.toml
index 9754213..4c18d5d 100755
--- a/Project.toml
+++ b/Project.toml
@@ -29,7 +29,7 @@ CategoricalArrays = "0.10"
 ChainRules = "1.49.0"
 ComputationalResources = "0.3"
 Distributions = "0.25.109"
-Flux = "0.13, 0.14"
+Flux = "0.13, 0.14, 0.16"
 InferOpt = "0.6.1"
 LaplaceRedux = "1.0.2"
 LinearAlgebra = " 1.9, 1.10"