I don’t think that tracks with all the exercise they show him doing in prison before then. Or the many comments about his shape before that. I feel like the implied arc is that he went downhill after losing his son and finally does a comeback to break out and meet up with the White Lotus and fulfill his destiny to take back Ba Sing Se.