fixup: fix broken tests