diff --git a/pybamm/expression_tree/symbol.py b/pybamm/expression_tree/symbol.py
index 7b6e330325..ace8f3f620 100644
--- a/pybamm/expression_tree/symbol.py
+++ b/pybamm/expression_tree/symbol.py
@@ -8,6 +8,8 @@
 import sympy
 from anytree.exporter import DotExporter
 from scipy.sparse import csr_matrix, issparse
+from functools import cache
+from functools import cached_property
 
 import pybamm
 from pybamm.expression_tree.printing.print_name import prettify_print_name
diff --git a/pybamm/models/base_model.py b/pybamm/models/base_model.py
index b26591a064..ff769134b9 100644
--- a/pybamm/models/base_model.py
+++ b/pybamm/models/base_model.py
@@ -7,6 +7,7 @@
 import copy
 import casadi
 import numpy as np
+from functools import cached_property
 
 import pybamm
 from pybamm.expression_tree.operations.latexify import Latexify
diff --git a/pybamm/models/full_battery_models/base_battery_model.py b/pybamm/models/full_battery_models/base_battery_model.py
index 42a660d088..b0bb881652 100644
--- a/pybamm/models/full_battery_models/base_battery_model.py
+++ b/pybamm/models/full_battery_models/base_battery_model.py
@@ -4,6 +4,7 @@
 
 import pybamm
 import numbers
+from functools import cached_property
 
 
 class BatteryModelOptions(pybamm.FuzzyDict):
diff --git a/pybamm/models/full_battery_models/lithium_ion/electrode_soh.py b/pybamm/models/full_battery_models/lithium_ion/electrode_soh.py
index 30adec88ef..21d4ce8a04 100644
--- a/pybamm/models/full_battery_models/lithium_ion/electrode_soh.py
+++ b/pybamm/models/full_battery_models/lithium_ion/electrode_soh.py
@@ -3,6 +3,7 @@
 #
 import pybamm
 import numpy as np
+from functools import cache
 
 
 class ElectrodeSOH(pybamm.BaseModel):
diff --git a/pybamm/simulation.py b/pybamm/simulation.py
index 0a55049a36..69d73928c3 100644
--- a/pybamm/simulation.py
+++ b/pybamm/simulation.py
@@ -7,6 +7,7 @@
 import copy
 import warnings
 import sys
+from functools import cache
 
 
 def is_notebook():
diff --git a/pybamm/solvers/solution.py b/pybamm/solvers/solution.py
index e9e35b2f60..2f6c219773 100644
--- a/pybamm/solvers/solution.py
+++ b/pybamm/solvers/solution.py
@@ -9,6 +9,7 @@
 import pybamm
 import pandas as pd
 from scipy.io import savemat
+from functools import cached_property
 
 
 class NumpyEncoder(json.JSONEncoder):