fixed classification bug