Change .unwrap to .expect.