From beeaab2eda55316f4d606a18599d070a52bb6906 Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@gmail.com>
Date: Wed, 31 Jul 2024 13:05:10 +0530
Subject: [PATCH] fix: fix incorrect function calls in linearization_function

---
 src/systems/abstractsystem.jl | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/systems/abstractsystem.jl b/src/systems/abstractsystem.jl
index e8f8f26e99..eee856f897 100644
--- a/src/systems/abstractsystem.jl
+++ b/src/systems/abstractsystem.jl
@@ -2172,9 +2172,8 @@ function linearization_function(sys::AbstractSystem, inputs,
             u_getter = u_getter
 
             function (u, p, t)
-                state = ProblemState(; u, p, t)
-                p_setter!(oldps, p_getter(state))
-                newu = u_getter(state)
+                p_setter!(oldps, p_getter(u, p..., t))
+                newu = u_getter(u, p, t)
                 return newu, oldps
             end
         end