diff --git a/test/Project.toml b/test/Project.toml
index 5751c756..8df22d17 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -29,7 +29,7 @@ LogDensityProblems = "0.12, 1, 2"
 LogExpFunctions = "0.3"
 MLJBase = "0.19, 0.20, 0.21"
 MLJDecisionTreeInterface = "0.3, 0.4"
-MLJIteration = "0.5"
+MLJIteration = "0.5, 0.6"
 MLJLIBSVMInterface = "0.2"
 MLJModels = "0.16"
 MLJXGBoostInterface = "0.3"