1 files changed, 3 insertions, 2 deletions
diff --git a/fswidget.h b/fswidget.h
index 236f6e0..61cb732 100644
--- a/fswidget.h
+++ b/fswidget.h
@@ -10,11 +10,12 @@ class QStandardItemModel;
 class QSortFilterProxyModel;
 class QContextMenuEvent;
 class NewMovieWizard;
+class FSProxy;
 
 class FSWidget : public QWidget {
     Q_OBJECT
     public:
-        enum CustomRoles { FullPathRole = Qt::UserRole + 1};
+        enum CustomRoles { FullPathRole = Qt::UserRole + 1, DurationRole = Qt::UserRole + 2, SizeRole = Qt::UserRole + 3 };
         explicit FSWidget(QWidget *parent = nullptr);
         ~FSWidget();
 
@@ -44,7 +45,7 @@ class FSWidget : public QWidget {
         QComboBox *mFilterCB;
         QTreeView *mFileView;
         QStandardItemModel *mModel;
-        QSortFilterProxyModel *mProxy;
+        FSProxy *mProxy;
         NewMovieWizard *mMovieWizard;
         int mQueryCount;
 };