I think this would work better as a `case` statement