Improve UnitAnnotatorTest