fix test breaking change