diff --git a/Project.toml b/Project.toml
index 075d4148..f99c1503 100644
--- a/Project.toml
+++ b/Project.toml
@@ -59,7 +59,7 @@ DistributedFactorGraphs = "0.25"
 Distributions = "0.24, 0.25"
 DocStringExtensions = "0.8, 0.9"
 FileIO = "1"
-Flux = "0.14"
+Flux = "0.14, 0.16"
 ImageCore = "0.9, 0.10"
 ImageIO = "0.6"
 IncrementalInference = "0.35"