diff --git a/Project.toml b/Project.toml
index be860f42..4b412c17 100644
--- a/Project.toml
+++ b/Project.toml
@@ -31,7 +31,7 @@ DiffEqCallbacks = "2.20.1"
 Functors = "0.2, 0.3, 0.4"
 LinearSolve = "1"
 Lux = "0.4"
-MLUtils = "0.4"
+MLUtils = "0.2, 0.3, 0.4"
 OrdinaryDiffEq = "6"
 SciMLBase = "1.19"
 SciMLSensitivity = "7"