Update the unit test to reflect the attribute name change