diff --git a/Project.toml b/Project.toml
index dc8b9e5..1f5eca8 100644
--- a/Project.toml
+++ b/Project.toml
@@ -29,7 +29,7 @@ SafeTestsets = "0.1"
 SciMLStructures = "1.1.0"
 StableRNGs = "1"
 SymbolicIndexingInterface = "0.3.15"
-Symbolics = "5.27, 6.0"
+Symbolics = "6.9"
 Test = "1.10"
 julia = "1.10"
 
diff --git a/docs/Project.toml b/docs/Project.toml
index 92889db..a00bb49 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -14,7 +14,7 @@ ModelingToolkitNeuralNets = "f162e290-f571-43a6-83d9-22ecc16da15f"
 
 [compat]
 Documenter = "1.3"
-Lux = "0.5.32"
+Lux = "1"
 ModelingToolkit = "9.9"
 ModelingToolkitStandardLibrary = "2.7"
 Optimization = "3.24"
diff --git a/src/utils.jl b/src/utils.jl
index 5489de1..a576ea6 100644
--- a/src/utils.jl
+++ b/src/utils.jl
@@ -1,6 +1,6 @@
 function multi_layer_feed_forward(input_length, output_length; width::Int = 5,
-        depth::Int = 1, activation = tanh, disable_optimizations = false)
+        depth::Int = 1, activation = tanh)
     Lux.Chain(Lux.Dense(input_length, width, activation),
         [Lux.Dense(width, width, activation) for _ in 1:(depth)]...,
-        Lux.Dense(width, output_length); disable_optimizations)
+        Lux.Dense(width, output_length))
 end