diff --git a/Project.toml b/Project.toml
index 1426cea7f5..ebfab91127 100644
--- a/Project.toml
+++ b/Project.toml
@@ -31,7 +31,7 @@ ArrayInterface = "3.1, 4, 5, 6"
 CUDA = "3"
 ChainRulesCore = "1.12"
 Functors = "0.3"
-MLUtils = "0.2, 0.3"
+MLUtils = "0.2, 0.3, 0.4"
 MacroTools = "0.5"
 NNlib = "0.8.9"
 NNlibCUDA = "0.2.4"