The ref did a poor job of cleaning it up early. He should have been giving lot of verbal warnings & maybe even an Unnecessary Roughness.

The goal is clean things up to prevent things escalating. You allow one wrestler to hit, the other is going to retaliate, usually harder, now you are in a situation where you have to make a major infraction call & you don't want that.