U: fix classification error