Fix test broken by 5082f6e587