1 files changed, 4 insertions, 2 deletions
diff --git a/src/previewgenerator.h b/src/previewgenerator.h
index e872b06b..0d8f0781 100644
--- a/src/previewgenerator.h
+++ b/src/previewgenerator.h
@@ -26,10 +26,12 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 #include <functional>
 #include <ipluginpreview.h>
 
+class PluginContainer;
+
 class PreviewGenerator
 {
 public:
-  PreviewGenerator();
+  PreviewGenerator(const PluginContainer* pluginContainer);
 
   void registerPlugin(MOBase::IPluginPreview *plugin);
 
@@ -44,8 +46,8 @@ private:
 
 private:
 
+  const PluginContainer* m_PluginContainer;
   std::map<QString, MOBase::IPluginPreview*> m_PreviewPlugins;
-
   QSize m_MaxSize;
 
 };