Even more trying to make the trainer agnostic to tagger