Ehm, yes. My mistake, I was confusing `y_classes` and `classes`.