diff --git a/YACReaderLibrary/trayicon_controller.cpp b/YACReaderLibrary/trayicon_controller.cpp
index 208d6b1c..601b98ad 100644
--- a/YACReaderLibrary/trayicon_controller.cpp
+++ b/YACReaderLibrary/trayicon_controller.cpp
@@ -2,6 +2,8 @@
 
 #include "yacreader_global_gui.h"
 
+#include "library_window.h"
+
 #include <QtWidgets>
 #include <QMessageBox>
 
@@ -11,7 +13,7 @@
 
 using namespace YACReader;
 
-TrayIconController::TrayIconController(QSettings *settings, QMainWindow *window)
+TrayIconController::TrayIconController(QSettings *settings, LibraryWindow *window)
     : QObject(nullptr), settings(settings), window(window)
 {
 
diff --git a/YACReaderLibrary/trayicon_controller.h b/YACReaderLibrary/trayicon_controller.h
index 649121ba..1ff9d407 100644
--- a/YACReaderLibrary/trayicon_controller.h
+++ b/YACReaderLibrary/trayicon_controller.h
@@ -4,6 +4,8 @@
 #include <QtWidgets>
 #include <QSystemTrayIcon>
 
+class LibraryWindow;
+
 namespace YACReader {
 
 class TrayIconController : public QObject
@@ -11,7 +13,7 @@ class TrayIconController : public QObject
     Q_OBJECT
 public:
     TrayIconController(QSettings *settings,
-                       QMainWindow *window);
+                       LibraryWindow *window);
 
     void updateIconVisibility();
 
@@ -24,7 +26,7 @@ public slots:
 
 private:
     QSettings *settings;
-    QMainWindow *window;
+    LibraryWindow *window;
 
     QMenu *trayIconMenu;
 };