Normally you would expect that to be the case, especially if they had brought someone like in Harris instead of Green.
But Aus have been searching for a genuine all-rounder since the dawn of time, which increases Green's stock quite a bit.
There was a lot of love for Burns after the match , particularly from JL, so I think he would stay at least for the next match.
At the end of the day I would expect it pans out as you surmised, but I'm half expecting a shock announcement.