fix class name in import