Fix test breakage: .ok().unwrap() instead of .unwrap()