TreeSet fixes