Bugfix: class labels correspond across tasks