Australia takes back the lead