Colombo: Australia captain Steve Smith is to leave the tour of Sri Lanka and return home to rest ahead of their limited-overs series in South Africa before they host the Proteas at the start of a heavy home schedule.
Vice-captain David Warner will lead the side for the remaining three one-day internationals and two Twenty20 matches against Sri Lanka.
Australia were hammered 3-0 in the test series, finding it difficult on the turning pitches in Sri Lanka, but achieved a three-wicket win in the first one-day international in Colombo.
Fast bowler Nathan Coulter-Nile has also been ruled out of the remaining three ODIs in the Sri Lanka tour with a lower back injury and will not be replaced in the squad, Cricket Australia said in a statement.
Peter Nevill, Australia's wicketkeeper for tests, will replace Matthew Wade for the two T20 matches with batsman Chris Lynn and all-rounder Glenn Maxwell also joining the 13-man squad for the shortest format.
T20 squad: David Warner (captain), Scott Boland, James Faulkner, Aaron Finch, John Hastings, Travis Head, Moises Henriques, Chris Lynn, Shaun Marsh, Glenn Maxwell, Peter Nevill (wk), Mitchell Starc, Adam Zampa