diff --git a/Project.toml b/Project.toml
index caec6dfd..10b9b625 100644
--- a/Project.toml
+++ b/Project.toml
@@ -52,7 +52,7 @@ Hyperopt = "0.5.6"
 JLD2 = "0.5.13, 0.6"
 Lux = "1.12.4"
 LuxCore = "1.2.4"
-MLJ = "0.20"
+MLJ = "0.20, 0.21"
 MLUtils = "0.4.8"
 Makie = "0.22, 0.23, 0.24"
 NCDatasets = "0.14.8"