1 files changed, 3 insertions, 3 deletions
diff --git a/archiveview.h b/archiveview.h
index a8aed6b..d97ca7f 100644
--- a/archiveview.h
+++ b/archiveview.h
@@ -9,9 +9,9 @@
 #define ARCHIVEVIEW_H
 
 #include <QWidget>
-#include <QTreeView>
 
 #include "archivemodel.h"
+#include "smtreeview.h"
 
 class QSortFilterProxyModel;
 class QComboBox;
@@ -33,7 +33,7 @@ class ArchiveView : public QWidget {
         ArchiveModel *mArchiveModel;
 };
 
-class ArchiveTree : public QTreeView {
+class ArchiveTree : public SmTreeView {
     public:
         explicit ArchiveTree(QWidget *parent = 0);
 
@@ -42,7 +42,7 @@ class ArchiveTree : public QTreeView {
         ArchiveModel *mModel;
 };
 
-class ArchiveFiles : public QTreeView {
+class ArchiveFiles : public SmTreeView {
     public:
         explicit ArchiveFiles(QWidget *parent = 0);
 };