diff --git a/lib/DataDrivenLux/Project.toml b/lib/DataDrivenLux/Project.toml
index 0cef66ef..b7a35e06 100644
--- a/lib/DataDrivenLux/Project.toml
+++ b/lib/DataDrivenLux/Project.toml
@@ -46,7 +46,7 @@ InverseFunctions = "0.1"
 Lux = "1"
 LuxCore = "1"
 Optim = "1.7"
-Optimisers = "0.3"
+Optimisers = "0.3, 0.4"
 ProgressMeter = "1.7"
 Setfield = "1"
 StatsBase = "0.34.3"