diff --git a/Project.toml b/Project.toml
index 8ae4d56..60a94e5 100644
--- a/Project.toml
+++ b/Project.toml
@@ -18,7 +18,7 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [compat]
 ChainRulesCore = "1"
-Flux = "0.15.2"
+Flux = "0.15.2, 0.16"
 Functors = "0.2, 0.3, 0.4, 0.5"
 JuMP = "0.19, 0.20, 0.21, 0.22, 0.23, 1"
 NaiveNASlib = "2.1"