fix test breakage