MDL-45258 tags: Fix issues with context related to tag_set() apis