fix class name update