Fix classifying 0 tokens