Fix test break.