Correction for NB confusion matrix