Head coach Brendon McCullum confirms Ben Stokes is set to return as England captain for the third Test against New Zealand.