Update code to use unwrap method.