Confusion matrix fix