Fix class name and imports.