Fixed class_weight_label problem