Fix class_name