Darren Lehmann will join Australia's squad in New Zealand on Saturday but will not resume full-time duties as national coach until the Test leg of the tour.
Lehmann be with the side for the second one-day international in Wellington but will not travel with the side for the final game in Hamilton. He will instead stay in Wellington, where the first Test is being played. Michael Di Venuto will remain as acting head coach.
Lehmann was on Thursday given the all clear by Cricket Australia's medical staff to fly after being diagnosed with deep vein thrombosis during the fifth one-day international against India in Sydney on January 23.
Perhaps coincidentally, Australia have not won a game since their coach's health scare, a point he has retweeted on social media.
Lehmann's return is a much-needed boost for an Australian squad still coming to grips with their heavy loss in the opening one-dayer. The loss was Australia's fifth in a row in the limited-overs formats, prompting captain Steve Smith to urge his players to lift.
"It was mainly player led. The coaches sort of took a back seat. Steve just said we need to improve in a number of areas," paceman John Hastings said.
"In the field we were a little bit sloppy so if we can tidy that up - look a bit more energetic out there. Obviously not losing clumps of wickets at the start, that's a pretty big key as well.
"They were some of the sorts of things he talked about, but it was really about just putting that one to bed, coming out and playing as best we can in the next couple."
Meanwhile, George Bailey has distanced himself from claims he was responsible for directing David Warner not to review an incorrect lbw decision against him.
Warner's dismissal was part of a dramatic batting collapse though it is debateable what effect it would have had on the game had he been given a reprieve.
"I said 'what does it feel like?'" Bailey told RSN Breakfast on Thursday.
"And he said 'high'.
"And I said 'go for it' and he turned around and walked off, so I don't really know what more he wants from my end."
There were reports alleging Bailey had referred to Warner as a "fool" but he had instead been calling the vice-captain by his nickname of "Bull". There are no suggestions of any rift between the two.