fix classifier bug