Ed and show both knew he'd be leaving at a certain point. That was clearly planned out. Best case scenario is he knew he had a meeting on a certain date he needed to make and as he was a late addition they made arrangements for him to go with the option to return and then Jill having ultimate say. I don't think she was at all shocked he would return. She's not stupid and knows how these things go. He kept the rose so even without spoilers I knew he'd be returning.