Fixed failing spec related to class name change.