Fix bugs in tag/untag