fix test break