Update tests for latest Rust