Free will, punishment and the right of society to defend itself.

People have suggested that if people lack free will, then they don't deserve to be punished. They seem to ignore that the people doing the punishment lack free will as well. I feel they also ask the wrong question, which is, do people deserve to be punished. A better question would be does society have the right to defend itself, and in what form should that defense take?