fixed classification issue