diff --git a/Project.toml b/Project.toml
index f2caf3b14..e17cf959a 100644
--- a/Project.toml
+++ b/Project.toml
@@ -23,7 +23,7 @@ CUDA = "3"
 ChainRulesCore = "1"
 Flux = "0.13"
 Functors = "0.2, 0.3"
-MLUtils = "0.2.10, 0.3"
+MLUtils = "0.2.10, 0.3, 0.4"
 NNlib = "0.8"
 NNlibCUDA = "0.2"
 PartialFunctions = "1"