diff --git a/idaes/models_extra/co2_capture_and_utilization/unit_models/tests/test_membrane_1d.py b/idaes/models_extra/co2_capture_and_utilization/unit_models/tests/test_membrane_1d.py
index ed07a55c5c..8c129ebffe 100644
--- a/idaes/models_extra/co2_capture_and_utilization/unit_models/tests/test_membrane_1d.py
+++ b/idaes/models_extra/co2_capture_and_utilization/unit_models/tests/test_membrane_1d.py
@@ -177,7 +177,7 @@ def test_build(self, membrane):
 
         assert hasattr(membrane.fs.unit, "mscontactor")
         assert hasattr(membrane.fs.unit, "permeability_calculation")
-        assert hasattr(membrane.fs.unit, "energy_transfer")
+        assert hasattr(membrane.fs.unit, "isothermal_constraint")
 
         assert number_variables(membrane) == 157
         assert number_total_constraints(membrane) == 89