Update tests for new Rust