diff --git a/openpype/tools/context_dialog/__init__.py b/openpype/tools/context_dialog/__init__.py
index 9b10baf903..15b90463da 100644
--- a/openpype/tools/context_dialog/__init__.py
+++ b/openpype/tools/context_dialog/__init__.py
@@ -1,10 +1,12 @@
-from .window import (
-    ContextDialog,
-    main
-)
+from openpype import AYON_SERVER_ENABLED
+
+if AYON_SERVER_ENABLED:
+    from ._ayon_window import ContextDialog, main
+else:
+    from ._openpype_window import ContextDialog, main
 
 
 __all__ = (
     "ContextDialog",
-    "main"
+    "main",
 )
diff --git a/openpype/tools/context_dialog/window.py b/openpype/tools/context_dialog/window.py
deleted file mode 100644
index 15b90463da..0000000000
--- a/openpype/tools/context_dialog/window.py
+++ /dev/null
@@ -1,12 +0,0 @@
-from openpype import AYON_SERVER_ENABLED
-
-if AYON_SERVER_ENABLED:
-    from ._ayon_window import ContextDialog, main
-else:
-    from ._openpype_window import ContextDialog, main
-
-
-__all__ = (
-    "ContextDialog",
-    "main",
-)