Fix test.rs