Williamson was injured and missed New Zealand's historic 3-0 series win over India in India last month