Fix test to reflect last change