Второй вариант с сортировкой, уже работает как надо.