1 files changed, 3 insertions, 2 deletions
diff --git a/searchdialog.cpp b/searchdialog.cpp
index 1f59839..8c255c8 100644
--- a/searchdialog.cpp
+++ b/searchdialog.cpp
@@ -15,6 +15,7 @@
 #include <QApplication>
 
 #include "searchdialog.h"
+#include "searchview.h"
 #include "helper.h"
 
 SearchDialog::SearchDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent, f){
@@ -39,7 +40,7 @@ SearchDialog::SearchDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent,
     mResM = new QStandardItemModel;
     QSortFilterProxyModel *resMProxy = new QSortFilterProxyModel;
     resMProxy->setSourceModel(mResM);
-    mResV = new QTreeView;
+    mResV = new SearchView;
     mResV->setModel(resMProxy);
     connect(mResV->selectionModel(), &QItemSelectionModel::currentChanged, this, &SearchDialog::doResult);
     QHBoxLayout *resGBL = new QHBoxLayout;
@@ -49,7 +50,7 @@ SearchDialog::SearchDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent,
     mDataM = new QStandardItemModel;
     QSortFilterProxyModel *dataMProxy = new QSortFilterProxyModel;
     dataMProxy->setSourceModel(mDataM);
-    mDataV = new QTreeView;
+    mDataV = new SearchView;
     mDataV->setModel(dataMProxy);
     QHBoxLayout *dataGBL = new QHBoxLayout;
     dataGBL->addWidget(mDataV);