Rename class in comment