Hitting back can include shoving away, and defensive blocks.
My minion was learning to use and moderate themselves at around that age. They started martial arts not long after. I would never punish or argue against proportionate self defense. Even if it's just to signal that it's not appropriate play, and an adult's attention is needed ASAP.