fix test broke