Fix test after changing class name display