IIRC, in the final scene, it was clear that he overcame the rehab and was back to his old, evil self.
He needed a lot of politically motivated and funded counter-therapy in order to do that.