Would a `switch` be better?