=== modified file 'dhis-2/dhis-web/dhis-web-commons-resources/src/main/webapp/dhis-web-commons/ajax/jsonOrganisationUnitTree.vm'
--- dhis-2/dhis-web/dhis-web-commons-resources/src/main/webapp/dhis-web-commons/ajax/jsonOrganisationUnitTree.vm	2011-07-24 15:07:45 +0000
+++ dhis-2/dhis-web/dhis-web-commons-resources/src/main/webapp/dhis-web-commons/ajax/jsonOrganisationUnitTree.vm	2011-07-25 09:32:54 +0000
@@ -6,7 +6,7 @@
 		"hasChildren": #if( $organisationUnit.children.size() > 0 )true,#{else}false#end
 		#if( $organisationUnit.children.size() > 0 )
 		"children": [
-			#set( $children = $sorter.sortOrganisationUnitSet( $organisationUnit.children ) ) 
+			#set( $children = $sorter.sort( $organisationUnit.children ) ) 
 			#foreach ( $child in $children )
 				#expandOrganisationUnit( $child )#if( $velocityCount < $organisationUnit.children.size() ),#end
 			#end

=== modified file 'dhis-2/dhis-web/dhis-web-commons/src/main/java/org/hisp/dhis/commons/action/GetOrganisationUnitTreeAction.java'
--- dhis-2/dhis-web/dhis-web-commons/src/main/java/org/hisp/dhis/commons/action/GetOrganisationUnitTreeAction.java	2011-07-24 15:07:45 +0000
+++ dhis-2/dhis-web/dhis-web-commons/src/main/java/org/hisp/dhis/commons/action/GetOrganisationUnitTreeAction.java	2011-07-25 09:32:54 +0000
@@ -88,20 +88,27 @@
         return organisationUnits;
     }
 
-    public class SetSorter
+    public class SetSorter<T>
     {
-        public Set<OrganisationUnit> sortOrganisationUnitSet( Set<OrganisationUnit> organisationUnits )
-        {
-            SortedSet<OrganisationUnit> sortedSet = new TreeSet<OrganisationUnit>( organisationUnitComparator );
-            sortedSet.addAll( organisationUnits );
+        private Comparator<T> comparator;
+
+        public SetSorter( Comparator<T> comparator )
+        {
+            this.comparator = comparator;
+        }
+
+        public Set<T> sort( Set<T> unsortedSet )
+        {
+            SortedSet<T> sortedSet = new TreeSet<T>( comparator );
+            sortedSet.addAll( unsortedSet );
 
             return sortedSet;
         }
     }
 
-    public SetSorter getSorter()
+    public SetSorter<OrganisationUnit> getSorter()
     {
-        return new SetSorter();
+        return new SetSorter<OrganisationUnit>( organisationUnitComparator );
     }
 
     // -------------------------------------------------------------------------