Update for rust changes.