Fixup
Merge pull request #165 from NielsZeilemaker/fix_unittest_annotations

Fixup