Classification error fix