fix pb with remove_label