diff --git a/test/Project.toml b/test/Project.toml
index ad2ebb0a..6f3f446b 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -26,7 +26,7 @@ FFTW = "1.1"
 LogDensityProblems = "0.12, 1, 2"
 LogExpFunctions = "0.3"
 MLJBase = "0.19, 0.20, 0.21"
-MLJDecisionTreeInterface = "0.3"
+MLJDecisionTreeInterface = "0.3, 0.4"
 MLJIteration = "0.5"
 MLJLIBSVMInterface = "0.2"
 MLJModels = "0.16"