England fast bowler Josh Tongue will return to the UK for assessment after suffering a toe injury ahead of the white-ball series against Australia.