Fixed confusion matrix bug