Fix #2 - sort the correct direction