What about just making them not retaliate when they’re “fleeing”?
They will only retaliate when you first hit them, and after they have calmed down.
Basically, when they’re still calm, they will retaliate as soon as you hit them and activate their “fleeing” state.
When you hit them in their fleeing state, they will not dare to fight back.