diff --git a/docs/Project.toml b/docs/Project.toml
index 490f6d1..f92ae8d 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -20,8 +20,8 @@ SymbolicIndexingInterface = "2efcf032-c050-4f8e-a9bb-153293bab1f5"
 SymbolicRegression = "8254be44-1295-4e6a-a16d-46603ac705cb"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
-[sources]
-ModelingToolkitNeuralNets = {path = ".."}
+[sources.ModelingToolkitNeuralNets]
+path = ".."
 
 [compat]
 Documenter = "1.3"
@@ -30,6 +30,7 @@ ModelingToolkit = "10"
 ModelingToolkitNeuralNets = "2"
 ModelingToolkitStandardLibrary = "2.7"
 Optimization = "4.0"
+OptimizationOptimJL = "0.4"
 OptimizationOptimisers = "0.3"
 Plots = "1"
 SciMLStructures = "1.1.0"