Fixed regression caused by the labels