diff --git a/Project.toml b/Project.toml
index 9754213..9f537ec 100755
--- a/Project.toml
+++ b/Project.toml
@@ -35,7 +35,7 @@ LaplaceRedux = "1.0.2"
 LinearAlgebra = " 1.9, 1.10"
 MLJBase = "0.20, 0.21, 1"
 MLJEnsembles = "0.3.3, 0.4"
-MLJFlux = "0.2.10, 0.3, 0.4, 0.5"
+MLJFlux = "0.2.10, 0.3, 0.4, 0.5, 0.6"
 MLJLinearModels = "0.10.0"
 MLJModelInterface = "1"
 MLUtils = "0.4.2"