diff --git a/qt.py b/qt.py
index 0d0cbaa..ccdcd95 100644
--- a/qt.py
+++ b/qt.py
@@ -83,8 +83,7 @@ class Plugin(BalPlugin):
                         title=_("Success"),
                     )
                     return
-                top_level_window= window.top_level_window
-                w = BalWindow(self, top_level_window)
+                w = BalWindow(self, window)
                 self.bal_windows[top_level_window.winId] = w
                 for child in window.children():
                     if isinstance(child, QMenuBar):
@@ -152,9 +151,8 @@ class Plugin(BalPlugin):
     def get_window(self, window):
         w = self.bal_windows.get(window.winId, None)
         if w is None:
-            win=w.top_level_window()
-            w = BalWindow(self, win)
-            self.bal_windows[win.winId] = w
+            w = BalWindow(self, window)
+            self.bal_windows[window.winId] = w
         return w
 
     def requires_settings(self):