Fix test broken by the previous commit