fix test broken by last change