Yes, a batsman can return to the field after retiring injured as soon as a wicket falls. If he is unable to return after all of the other batters on the batting team have been dismissed, the team is declared all out and the play continues.