diff --git a/pype/modules/base.py b/pype/modules/base.py
index a320fd797e..bb166abca1 100644
--- a/pype/modules/base.py
+++ b/pype/modules/base.py
@@ -278,13 +278,11 @@ class TrayModulesManager(ModulesManager):
         self.modules_by_id = {}
         self.modules_by_name = {}
 
-    def initialize(self, tray_widget, main_window):
-        self.tray_widget = tray_widget
-        self.main_window = main_window
-
+    def initialize(self, tray_menu):
         self.initialize_modules()
         self.tray_init()
         self.connect_modules()
+        self.tray_menu(tray_menu)
 
     def get_enabled_tray_modules(self):
         output = []
diff --git a/pype/tools/tray/pype_tray.py b/pype/tools/tray/pype_tray.py
index c931e819a8..55f8e975db 100644
--- a/pype/tools/tray/pype_tray.py
+++ b/pype/tools/tray/pype_tray.py
@@ -36,9 +36,7 @@ class TrayManager:
     def initialize_modules(self):
         """Add modules to tray."""
 
-        self.modules_manager.initialize(self.tray_widget, self.main_window)
-
-        self.modules_manager.tray_menu(self.tray_widget.menu)
+        self.modules_manager.initialize(self.tray_widget.menu)
 
         # Add services if they are
         services_submenu = ITrayService.services_submenu()