Update tests after class name changes.