rename class
small fix