fix class name on comments