fixup, add test