Make players select augments when they load into the match, and not at hero selection (just like crests).
This is a very minimal difference, but it allows you to pick what augment is best against your opposing laner.
I think this opens the door for more thoughtful counter-play in certain situations where you're unsure who is your opposing laner.