diff --git a/Project.toml b/Project.toml
index 48418ef4ac..248dd074d3 100644
--- a/Project.toml
+++ b/Project.toml
@@ -28,7 +28,7 @@ Adapt = "3.0"
 CUDA = "3"
 ChainRulesCore = "1.12"
 Functors = "0.3, 0.4"
-MLUtils = "0.2, 0.3.1"
+MLUtils = "0.2, 0.3.1, 0.4"
 MacroTools = "0.5"
 NNlib = "0.8.9"
 NNlibCUDA = "0.2.4"