I think an `if else` would clearer here