I've seen it suggested elsewhere online that Jon was aware of Arya trying to also get past the zombie dragon and he was trying to distract it. He is shouting " Go!" When he confronts it and the next shot is the draft of Arya sprinting past the White Walker.