In hindsight, you're right. Either way should work with the latter being more clear.