fixes test breakages

/cc @densh @VladUreche