I think that would be a good refactoring idea if it gets any more complex, but I think the current approach is more succinct given that we're only worried about two scenarios.