I actually really disliked the direction the show went in this episode. It was so heavily telegraphed that she was missing being a teacher for a lot of the season. I would have preferred if they were teeing her up to learn that she could help way more kids if she kept advocating for them at the district level. The show kept crowbarring her into episodes so it's not like she was taking a job in another city or anything.
Heck, if she was going to end up back at Abbott, I would have preferred if her offer got revoked for breaking district policy to get that kid back to class.
I think it was exacerbated by my generally not liking the episode. The Bear parody was flimsy. Questlove bit was telegraphed. Melissa's bit almost killed the episode for me. I hope that never comes back.