Really fix the labels assumption!