fix the classification problem