diff --git a/watertap/examples/flowsheets/electrodialysis/tests/test_electrodialysis_1stack.py b/watertap/examples/flowsheets/electrodialysis/tests/test_electrodialysis_1stack.py
index b63f72a255..434bc16501 100644
--- a/watertap/examples/flowsheets/electrodialysis/tests/test_electrodialysis_1stack.py
+++ b/watertap/examples/flowsheets/electrodialysis/tests/test_electrodialysis_1stack.py
@@ -47,10 +47,7 @@
     assert_degrees_of_freedom,
 )
 
-from watertap.costing.watertap_costing_package import (
-    WaterTAPCosting,
-    make_capital_cost_var,
-)
+from watertap.costing import WaterTAPCosting
 from watertap.property_models.multicomp_aq_sol_prop_pack import MCASParameterBlock
 
 import watertap.examples.flowsheets.electrodialysis.electrodialysis_1stack as edfs