Unwrap condition in comprehension