You're right. This doesn't belong here. It should be checked in plot_confusion().