Modify class name in comment