From 5921fba2c7039a7135f2a7e0371d2a0d2f23ce2c 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 | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/systems/abstractsystem.jl b/src/systems/abstractsystem.jl
index e8f8f26e99..9143e7a751 100644
--- a/src/systems/abstractsystem.jl
+++ b/src/systems/abstractsystem.jl
@@ -2173,8 +2173,8 @@ function linearization_function(sys::AbstractSystem, inputs,
 
             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