We should upgrade our UI tests to Rust 2021 This mostly involves adding a `// edition:2021` up top and ensuring the test still passes. It's worth splitting this task up.