diff --git a/Project.toml b/Project.toml
index 30fe75d..9172b47 100644
--- a/Project.toml
+++ b/Project.toml
@@ -39,7 +39,7 @@ Distributions = "0.25"
 Flux = "0.13, 0.14"
 KernelDensity = "0.6"
 MLJBase = "0.21"
-MLJFlux = "0.2"
+MLJFlux = "0.2, 0.3"
 MLJModelInterface = "1.3"
 ModelingToolkit = "8.0"
 Optimization = "3.5"