fix the classification problem
merge