Fix test broken at the last minute