That's a fair point. I would argue that we're not trying to optimize time in prison but benefit to society. Your argument relies on a linear(ish) relationship between the two.