Fix num_classes instead of num_labels in test_utilities