Fixed class_weight issue