Rust update: Fix tests