fix for rust changes