Fix build for Rust master