From a4c0030a51ab17162286074a493e58592e7cc199 Mon Sep 17 00:00:00 2001
From: NicolaCourtier <45851982+NicolaCourtier@users.noreply.github.com>
Date: Wed, 7 Aug 2024 15:57:18 +0100
Subject: [PATCH] Update test_models.py

---
 tests/unit/test_models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/test_models.py b/tests/unit/test_models.py
index e4ba3137..c96b10b4 100644
--- a/tests/unit/test_models.py
+++ b/tests/unit/test_models.py
@@ -143,7 +143,7 @@ def test_predict_without_allow_infeasible_solutions(self, model):
     @pytest.mark.unit
     def test_build(self, model):
         if isinstance(model, pybop.lithium_ion.SPMe):
-            model.build(init_soc=1.0)
+            model.build(initial_state={"Initial SoC": 1.0})
 
             # Test attributes with init_soc
             assert model.built_model is not None