fix test broken