Updated use to SortedSet to match TSK change