diff --git a/scripts/steady_state_demo.py b/scripts/steady_state_demo.py
index 45cc7c7..f030099 100644
--- a/scripts/steady_state_demo.py
+++ b/scripts/steady_state_demo.py
@@ -33,9 +33,7 @@ def get_steady_state_from_params(parameters: PyTree):
 
             This lets us get the Jacobian wrt (just) the parameters.
             """
-            _model = RateEquationModel(
-                parameters, model.structure, model.rate_equations
-            )
+            _model = RateEquationModel(parameters, model.structure)
             return get_kinetic_model_steady_state(_model, guess)
 
         # solve once for jitting