fixup! add tests