修正取分类的逻辑