1 files changed, 4 insertions, 3 deletions
diff --git a/filesorter.cpp b/filesorter.cpp
index a3bbf5a..aee6177 100644
--- a/filesorter.cpp
+++ b/filesorter.cpp
@@ -1,12 +1,13 @@
 #include "filesorter.h"
+#include "torrentwidget.h"
 #include "filewidget.h"
 
 FileSorter::FileSorter(QObject *parent) : QSortFilterProxyModel(parent) {}
 
 bool FileSorter::lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const {
-    QVariant leftData = sourceModel()->data(source_left, Qt::UserRole + 1);
-    QVariant rightData = sourceModel()->data(source_right, Qt::UserRole + 1);
-    if(leftData.type() == QVariant::Int){
+    if(source_left.column() == TorrentWidget::IconColumn){
+        QVariant leftData = sourceModel()->data(source_left, TorrentWidget::PresentRole);
+        QVariant rightData = sourceModel()->data(source_right, TorrentWidget::PresentRole);
         return leftData.toInt() < rightData.toInt();
     }
     return QSortFilterProxyModel::lessThan(source_left, source_right);