Let 'em take it before the attack action. I'm fine with trying it to the attack action, but it should be able to be taken prior to rolling the attack.
"When you take the attack action, prior to rolling the attack, you can attempt to shove as a bonus action."
Works for me.