1 files changed, 3 insertions, 7 deletions
diff --git a/mappingtablewidget.h b/mappingtablewidget.h
index 7c9e893..787d7f6 100644
--- a/mappingtablewidget.h
+++ b/mappingtablewidget.h
@@ -13,12 +13,8 @@
 #include <QIcon>
 
 #include "smdialog.h"
+#include "smtreeview.h"
 
-class QTreeView;
-class QLineEdit;
-class QPushButton;
-class QLabel;
-class QComboBox;
 class MappingTableItemModel;
 class MappingTableModel;
 
@@ -39,7 +35,7 @@ class MappingTableWidget : public QWidget{
 		void removeItem();
 
 	private:
-		QTreeView *mView;
+        SmTreeView* mView;
 		QPushButton *mAddItem;
 		QPushButton *mRemoveItem;
 		QPushButton *mSearchItem;
@@ -79,7 +75,7 @@ class MappingTableResultDialog : public SmDialog {
 		void accept();
 
 	private:
-		QTreeView *mView;
+        SmTreeView *mView;
 		QStringListModel *mModel;
 		QPushButton *mOk;
 		QPushButton *mCancel;