diff --git a/example/Burgers/Project.toml b/example/Burgers/Project.toml
index 2b6f9a1..0038d67 100644
--- a/example/Burgers/Project.toml
+++ b/example/Burgers/Project.toml
@@ -15,7 +15,7 @@ NeuralOperators = "ea5c82af-86e5-48da-8ee1-382d6ad7af4b"
 BSON = "0.3"
 CUDA = "3.12, 4"
 DataDeps = "0.7"
-Flux = "0.13"
+Flux = "0.13, 0.14"
 FluxTraining = "0.3"
 MAT = "0.10"
 MLUtils = "0.3"