1 files changed, 4 insertions, 0 deletions
diff --git a/src/downloadlistwidget.h b/src/downloadlistwidget.h
index 93ece07a..c8d4fe6e 100644
--- a/src/downloadlistwidget.h
+++ b/src/downloadlistwidget.h
@@ -21,6 +21,7 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 #define DOWNLOADLISTWIDGET_H
 
 #include "downloadmanager.h"
+#include "downloadlist.h"
 #include "downloadlistsortproxy.h"
 #include <QWidget>
 #include <QItemDelegate>
@@ -58,6 +59,8 @@ public:
   ~DownloadListWidget();
 
   void setManager(DownloadManager *manager);
+  void setSourceModel(DownloadList *sourceModel);
+  void setMetaDisplay(bool metaDisplay);
 
 signals:
   void installDownload(int index);
@@ -95,6 +98,7 @@ private slots:
 
 private:
   DownloadManager *m_Manager;
+  DownloadList *m_SourceModel;
   int m_ContextRow;
 };