Star New Zealand batter Kane Williamson will return from injury to take on England, replacing someone in the Test side that thrashed India on the subcontinent.