diff --git a/src/pybamm/solvers/base_solver.py b/src/pybamm/solvers/base_solver.py
index fbe9ac4b04..a43a710e7b 100644
--- a/src/pybamm/solvers/base_solver.py
+++ b/src/pybamm/solvers/base_solver.py
@@ -864,9 +864,9 @@ def solve(
                 # If the new initial conditions are different
                 # and cannot be evaluated directly, set up again
                 self.set_up(model, model_inputs_list[0], t_eval, ics_only=True)
-            self._model_set_up[model][
-                "initial conditions"
-            ] = model.concatenated_initial_conditions
+            self._model_set_up[model]["initial conditions"] = (
+                model.concatenated_initial_conditions
+            )
         else:
             # Set the standard initial conditions
             self._set_initial_conditions(model, t_eval[0], model_inputs_list[0])