diff --git a/example/DoublePendulum/Project.toml b/example/DoublePendulum/Project.toml
index 690c320b..94ea36c1 100644
--- a/example/DoublePendulum/Project.toml
+++ b/example/DoublePendulum/Project.toml
@@ -22,7 +22,7 @@ DataDeps = "0.7"
 DataFrames = "1.4"
 Flux = "0.13"
 FluxTraining = "0.3"
-MLUtils = "0.3"
+MLUtils = "0.3, 0.4"
 NeuralOperators = "0.4"
 Plots = "1.36"
 Pluto = "0.19"