Switch label_image to labels