Could you avoid the assignment here and just return the result of the `if` expression?