1 files changed, 3 insertions, 2 deletions
diff --git a/searchdialog.h b/searchdialog.h
index 0ef4aad..b5b9ba4 100644
--- a/searchdialog.h
+++ b/searchdialog.h
@@ -8,6 +8,7 @@ class QComboBox;
 class QTreeView;
 class QStandardItemModel;
 class QStandardItem;
+class SearchView;
 
 class SearchDialog : public QDialog {
     Q_OBJECT
@@ -37,8 +38,8 @@ class SearchDialog : public QDialog {
         void doActorGenres(QStandardItem *item);
         QLineEdit *mSearch;
         QComboBox *mTypeSel;
-        QTreeView *mResV;
-        QTreeView *mDataV;
+        SearchView *mResV;
+        SearchView *mDataV;
         QStandardItemModel *mResM;
         QStandardItemModel *mDataM;
 };