1 files changed, 6 insertions, 1 deletions
diff --git a/src/modflagicondelegate.h b/src/modflagicondelegate.h
index 63ab8978..24dd4a0e 100644
--- a/src/modflagicondelegate.h
+++ b/src/modflagicondelegate.h
@@ -6,12 +6,14 @@
 #include "icondelegate.h"
 #include "modinfo.h"
 
+class ModListView;
+
 class ModFlagIconDelegate : public IconDelegate
 {
   Q_OBJECT;
 
 public:
-  explicit ModFlagIconDelegate(QTreeView* view, int column = -1, int compactSize = 120);
+  explicit ModFlagIconDelegate(ModListView* view, int column = -1, int compactSize = 120);
   QSize sizeHint(const QStyleOptionViewItem &option, const QModelIndex &index) const override;
 
 protected:
@@ -25,6 +27,9 @@ protected:
   //
   ModFlagIconDelegate() : ModFlagIconDelegate(nullptr) { }
 
+private:
+  ModListView* m_view;
+
 };
 
 #endif // MODFLAGICONDELEGATE_H