Fixes classification labeling