fix Confusion bug

This PR is urgent as the ConfusionMatrix wont even load without it. Just a missing `end` clause.