Fix the unwrap optional error