fixup: fix unit tests.