Australia almost doubled their previous best total of the series against Sri Lanka on Monday, but it still wasn't enough to give them the upper hand in the third Test in Colombo.
Just before stumps on day three, the tourists were bowled out for 379 to give them a slender 24-run lead after the hosts made 355.
Australia's innings had much in common with Sri Lanka's as one massive record partnership provided the bulk of the runs at Sinhalese Sports Club.
A 246-run partnership between Shaun Marsh (130) and Steve Smith (119) lifted Australia to a dominant position but in familiar fashion for this tour, they stuttered against Sri Lanka's spinners.
At 1-267 on a wicket holding up surprisingly well after three hot days, Australia were well set to edge up towards the 500-run mark as they aim for their first Test win in the subcontinent since 2011.