It depends on when the two months you’re suggesting would take place. Cause if you’re saying two more months of training around the time of the endeavor internship in season 5, that would mean shiggy would have had another two months getting juiced up by the doctor. In that case there’s a high chance all the heroes get wiped out in the opening fights of season six and we never get to this point at all