diff --git a/Project.toml b/Project.toml
index 485646f..94d5556 100644
--- a/Project.toml
+++ b/Project.toml
@@ -21,7 +21,7 @@ CUDA = "3"
 DiffEqFlux = "1"
 DifferentialEquations = "6, 7"
 DomainSets = "0.5"
-Flux = "0.12"
+Flux = "0.12, 0.13"
 GalacticOptim = "2"
 ModelingToolkit = "6, 7, 8"
 NeuralPDE = "4"