diff --git a/tests/test_land.py b/tests/test_land.py index 0151f8c5..89e69340 100644 --- a/tests/test_land.py +++ b/tests/test_land.py @@ -24,7 +24,7 @@ from wsimod.nodes.nodes import Node from wsimod.nodes.sewer import Sewer from wsimod.nodes.storage import Reservoir -from wsimod.nodes.tanks import Tank +from wsimod.nodes.tanks import Tank, DecayTank from wsimod.orchestration.model import to_datetime from pathlib import Path