1 files changed, 4 insertions, 7 deletions
diff --git a/mappingtreewidget.h b/mappingtreewidget.h
index 1bd77f4..d72dffa 100644
--- a/mappingtreewidget.h
+++ b/mappingtreewidget.h
@@ -8,19 +8,16 @@
 #ifndef MAPPINGTREEWIDGET_H
 #define MAPPINGTREEWIDGET_H
 
-#include <QtWidgets/QWidget>
-#include <QtWidgets/QTreeView>
 #include <QtWidgets/QDialog>
 
+#include "smtreeview.h"
+
 class MappingTreeView;
 class MappingTreeModel;
 class QComboBox;
-class QPushButton;
-class QLabel;
 class QCheckBox;
 class QSortFilterProxyModel;
 class QStringListModel;
-class QAction;
 class MappingTreeResultModel;
 
 // defined in mappingtreemodel.h
@@ -66,7 +63,7 @@ class MappingTreeWidget : public QWidget {
         QAction *mMoveChildA;
 };
 
-class MappingTreeView : public QTreeView {
+class MappingTreeView : public SmTreeView {
     Q_OBJECT
     public:
         MappingTreeView(QWidget *parent = 0);
@@ -75,7 +72,7 @@ class MappingTreeView : public QTreeView {
         virtual void contextMenuEvent(QContextMenuEvent *e);
 };
 
-class MappingTreeResultView : public QTreeView {
+class MappingTreeResultView : public SmTreeView {
     Q_OBJECT
     public:
         explicit MappingTreeResultView(QWidget *parent = 0);