diff --git a/Project.toml b/Project.toml
index d60fa60..1e154ab 100644
--- a/Project.toml
+++ b/Project.toml
@@ -132,7 +132,7 @@ ExponentialUtilities = "1.18"
 FFTW = "1"
 FillArrays = "0.13, 1"
 FiniteDiff = "2.14"
-Flux = "0.13"
+Flux = "0.13, 0.16"
 ForwardDiff = "0.10.23 - 0.10.32"
 GLM = "1"
 GlobalSensitivity = "= 2.1.3"