If we believe prisons are to act as a sort of operant conditioning for the inmates, then the punishment IS supposed the rehabilitation. However, as you probably know, many folks do not respond to this type of conditioning.