Fix build

Fixes breakage caused by rust-lang/rust#18229 and rust-lang/rust#18365