1 files changed, 4 insertions, 5 deletions
diff --git a/collectionwidget.h b/collectionwidget.h
index 33799eb..bb85e49 100644
--- a/collectionwidget.h
+++ b/collectionwidget.h
@@ -2,7 +2,8 @@
 #define COLLECTIONWIDGET_H
 
 #include <QWidget>
-#include <QTreeView>
+
+#include "beetview.h"
 
 class QStandardItemModel;
 class QSortFilterProxyModel;
@@ -14,20 +15,18 @@ class CollectionWidget : public QWidget {
         enum CustomRoles { TypeRole = Qt::UserRole + 1, IdRole = Qt::UserRole + 2, FullPathRole = Qt::UserRole + 3, GenreRole = Qt::UserRole + 4, ArtistRole = Qt::UserRole + 5, TitleRole = Qt::UserRole + 6, AlbumRole = Qt::UserRole + 7, LengthRole = Qt::UserRole + 8, UrlRole = Qt::UserRole + 9, RemoteRole = Qt::UserRole + 10 };
         enum ItemType { Artist, Album, Song, Genre, WebRadio };
         explicit CollectionWidget(QWidget *parent = nullptr);
-        QTreeView *view() { return mView; }
+        BeetView *view() { return mView; }
         QStandardItemModel *model() { return mModel; }
         void setHeaders(const QStringList headers) { mHeaders = headers; }
         const QStringList headers() const { return mHeaders; }
         void disableSorting() { mView->setSortingEnabled(false); }
         void enableSorting(int column = 0, Qt::SortOrder sortOrder = Qt::AscendingOrder);
 
-    signals:
-
     public slots:
         virtual void populate() { return; }
 
     private:
-        QTreeView *mView;
+        BeetView *mView;
         QStandardItemModel *mModel;
         QLineEdit *mFilter;
         QStringList mHeaders;