I was going to suggest using variables to do this, but I think your solution is much better and works very well.
The only issue I had was creating the condition visually - clicking ‘select’ closed the data bindings panel without adding the code. I had to add it manually. Perhaps it was because the form was in a modal.