Lockie Ferguson will return to New Zealand after sustaining a calf injury during their series-levelling victory against Sri Lanka in the second T20I on Sunday.