diff --git a/Project.toml b/Project.toml
index 9b4c73c4..d6f236e6 100644
--- a/Project.toml
+++ b/Project.toml
@@ -30,7 +30,7 @@ ComputationalResources = "0.3.2"
 DataFrames = "1.5.0"
 Flux = "0.12, 0.13, 0.14"
 MLJ = "0.19.2"
-MLJBase = "0.21.11"
+MLJBase = "0.21.11, 1"
 MLJFlux = "0.2.10, 0.3"
 MLJModelInterface = "1.8.0"
 MLUtils = "0.4.3"