diff --git a/Project.toml b/Project.toml
index ee0e5a34..fe252bd8 100644
--- a/Project.toml
+++ b/Project.toml
@@ -40,7 +40,7 @@ DiffEqBase = "6"
 DiffEqUncertainty = "1"
 DifferentialEquations = "7"
 Distributions = "0.25"
-Flux = "0.13, 0.14"
+Flux = "0.13, 0.14, 0.16"
 ForwardDiff = "0.10"
 Franklin = "0.10"
 IntervalArithmetic = "0.22"