Rehabilitation. If neuroscience shows compulsion, punishment is just revenge in a lab coat; fix the driver before putting them back on the road.
Agreed, but rehab also has to grapple with consent and coercion—"fixing" someone's brain without their buy-in starts to look a lot like the punishment we're trying to escape.