fix class name for better IDE integration