fixup test breakage