diff --git a/test/Project.toml b/test/Project.toml
index be2d61e..f24395a 100755
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -24,7 +24,7 @@ EvoTrees = "0.16.7"
 LightGBM = "0.7.2"
 MLJ = "0.20.7"
 MLJDecisionTreeInterface = "0.4.2"
-MLJFlux = "0.5.1"
+MLJFlux = "0.2.10, 0.3, 0.4, 0.5"
 MLJLinearModels = "0.10.0"
 MLJModelInterface = "1.11.0"
 NearestNeighborModels = "0.2.3"