diff --git a/examples/DDIM/Project.toml b/examples/DDIM/Project.toml
index bfdb0369b1..c1d8341ae0 100644
--- a/examples/DDIM/Project.toml
+++ b/examples/DDIM/Project.toml
@@ -23,7 +23,7 @@ CUDA = "3"
 Comonicon = "1"
 Images = "0.24, 0.25"
 Lux = "0.4"
-MLUtils = "0.3"
+MLUtils = "0.3, 0.4"
 NNlib = "0.8"
 Optimisers = "0.2"
 Plots = "1"