modify the bug in classifier