Fix build break with rustc update