diff --git a/Project.toml b/Project.toml
index ad29b79..48afb0d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -22,7 +22,7 @@ IntervalSets = "0.7.10"
 JET = "0.8, 0.9"
 Lux = "1"
 LuxCore = "1"
-ModelingToolkit = "9.64"
+ModelingToolkit = "10"
 ModelingToolkitStandardLibrary = "2.7"
 Optimization = "3.24, 4"
 OptimizationOptimisers = "0.2.1, 0.3"
diff --git a/src/ModelingToolkitNeuralNets.jl b/src/ModelingToolkitNeuralNets.jl
index 3ce8533..d088a10 100644
--- a/src/ModelingToolkitNeuralNets.jl
+++ b/src/ModelingToolkitNeuralNets.jl
@@ -42,8 +42,8 @@ function NeuralNetworkBlock(; n_input = 1, n_output = 1,
 
     eqs = [output.u ~ out]
 
-    ude_comp = ODESystem(
-        eqs, t_nounits, [], [lux_model, p, T]; systems = [input, output], name)
+    ude_comp = System(
+        eqs, t_nounits, [lux_model, p, T]; systems = [input, output], name)
     return ude_comp
 end