McCullum 'anticipating' Stokes' return as captain for third Test