fix class names in comments