Update to match the name change of entity class