diff --git a/example/SuperResolution/Project.toml b/example/SuperResolution/Project.toml
index edaf10c4..1579a538 100644
--- a/example/SuperResolution/Project.toml
+++ b/example/SuperResolution/Project.toml
@@ -19,7 +19,7 @@ BSON = "0.3"
 CUDA = "3.12"
 Flux = "0.13"
 FluxTraining = "0.3"
-MLUtils = "0.3"
+MLUtils = "0.3, 0.4"
 NeuralOperators = "0.4"
 Plots = "1.36"
 Pluto = "0.19"