Kane Williamson flying back to NZ