fix remove_label exceptions