diff --git a/qt.py b/qt.py
index ccdcd95..0010b4a 100644
--- a/qt.py
+++ b/qt.py
@@ -83,7 +83,8 @@ class Plugin(BalPlugin):
                         title=_("Success"),
                     )
                     return
-                w = BalWindow(self, window)
+                top_level_window=window.top_level_window()
+                w = BalWindow(self, top_level_window)
                 self.bal_windows[top_level_window.winId] = w
                 for child in window.children():
                     if isinstance(child, QMenuBar):
@@ -149,6 +150,7 @@ class Plugin(BalPlugin):
         _logger.debug("daemon wallet loaded")
 
     def get_window(self, window):
+        window=window.top_level_window()
         w = self.bal_windows.get(window.winId, None)
         if w is None:
             w = BalWindow(self, window)