1 files changed, 2 insertions, 0 deletions
diff --git a/archiveview.h b/archiveview.h
index 7d477e8..7a288db 100644
--- a/archiveview.h
+++ b/archiveview.h
@@ -117,6 +117,7 @@ class ArchiveFiles : public SmTreeView {
     Q_OBJECT
     public:
         explicit ArchiveFiles(const QString &headerSettings = "afilesheaders", QWidget *parent = 0);
+        void setSourceModel(ArchiveFilesModel *m) { mSourceModel = m; }
 
     public slots:
         void deleteFromFiles();
@@ -127,6 +128,7 @@ class ArchiveFiles : public SmTreeView {
 
     private:
         QPoint mDragStartPos;
+        ArchiveFilesModel *mSourceModel;
 };
 
 class ArchiveProxy : public QSortFilterProxyModel {