diff --git a/client/ayon_core/tools/tray/__init__.py b/client/ayon_core/tools/tray/__init__.py
index d646880e15..9dbacc54c2 100644
--- a/client/ayon_core/tools/tray/__init__.py
+++ b/client/ayon_core/tools/tray/__init__.py
@@ -1,7 +1,10 @@
 from .webserver import HostMsgAction
 from .addons_manager import TrayAddonsManager
 from .lib import (
+    TrayState,
+    get_tray_state,
     is_tray_running,
+    get_tray_server_url,
     main,
 )
 
@@ -9,5 +12,10 @@ from .lib import (
 __all__ = (
     "HostMsgAction",
     "TrayAddonsManager",
+
+    "TrayState",
+    "get_tray_state",
+    "is_tray_running",
+    "get_tray_server_url",
     "main",
 )