Fix confusion matrix not being printed