diff --git a/Project.toml b/Project.toml
index 18b27649dc..3af67f51c4 100644
--- a/Project.toml
+++ b/Project.toml
@@ -85,7 +85,7 @@ FunctionWrappers = "1.1.3"
 Functors = "0.5"
 GPUArraysCore = "0.1.6, 0.2"
 LinearAlgebra = "1.10"
-LossFunctions = "0.11.1"
+LossFunctions = "0.11.1, 1"
 LuxCore = "1.2"
 LuxLib = "1.3.7"
 MLDataDevices = "1.6"