[refactor] Handle tags properly from existing flatiron API.