I'm generally not ok with physically punishing children... But in this case you're dealing with a kid who is using violence against someone else, and I think the punishment fits the crime. He's gotta learn that if you go around hitting people you're gonna get hit back...
Maybe the lesson he will learn, is that violence is an acceptable method of influencing peoples behaviour.