Update the unit test class to match the new domain object