bug fix: class name update.