diff --git a/Project.toml b/Project.toml
index 10bad812d..5f86abc9d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -51,6 +51,18 @@ SimpleChains = "de6bee2f-e2f4-4ec7-b6ed-219cc6f6e9e5"
 Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
+[sources.LuxCore]
+path = "lib/LuxCore"
+
+[sources.LuxLib]
+path = "lib/LuxLib"
+
+[sources.MLDataDevices]
+path = "lib/MLDataDevices"
+
+[sources.WeightInitializers]
+path = "lib/WeightInitializers"
+
 [extensions]
 LuxComponentArraysExt = "ComponentArrays"
 LuxEnzymeExt = "Enzyme"
@@ -79,7 +91,7 @@ DispatchDoctor = "0.4.12"
 Enzyme = "0.13.16"
 EnzymeCore = "0.8.6"
 FastClosures = "0.3.2"
-Flux = "0.15"
+Flux = "0.15, 0.16"
 ForwardDiff = "0.10.36"
 FunctionWrappers = "1.1.3"
 Functors = "0.5"
@@ -111,9 +123,3 @@ Tracker = "0.2.37"
 WeightInitializers = "1"
 Zygote = "0.6.70"
 julia = "1.10"
-
-[sources]
-LuxCore = { path = "lib/LuxCore" }
-LuxLib = { path = "lib/LuxLib" }
-MLDataDevices = { path = "lib/MLDataDevices" }
-WeightInitializers = { path = "lib/WeightInitializers" }