[Bugfix] change class name