fix class name reference