Update for Rust CI