Australia were trounced at Eden Park but coach Michael Cheika trusts his players will put the defeat behind them and focus on the World Cup.
Steve Hansen expects Richie Mo'unga to be available for New Zealand's Rugby World Cup campaign, dismissing worries over a shoulder injury.
Sri Lanka had victory in sight by stumps on day four as their record-setting openers starred in the first Test against New Zealand.
New Zealand atoned for defeat in Perth by thrashing Australia 36-0 in Auckland to retain the Bledisloe Cup, pleasing captain Kieran Read.
New Zealand scored five tries in a clinical Bledisloe Cup victory over an Australia side who humbled them seven days earlier in Perth.
New Zealand will be looking for BJ Watling to stretch New Zealand's lead beyond 200 when they resume on day four.
BJ Watling raised hopes of New Zealand winning the first Test after Sri Lanka had the tourists in the mire on Friday.
New Zealand fell to a stunning loss against Australia last time out, but All Blacks captain Kieran Read is confident they will bounce back.
New Zealand host Australia at fortress Eden Park looking for a first win in three Tests to retain the Bledisloe Cup.
Australia need to avoid defeat against New Zealand on Saturday to secure the Bledisloe Cup for the first time since 2002.