diff --git a/src/webdrop_bridge/ui/update_manager_ui.py b/src/webdrop_bridge/ui/update_manager_ui.py
index 7a45e13..1ddd4f0 100644
--- a/src/webdrop_bridge/ui/update_manager_ui.py
+++ b/src/webdrop_bridge/ui/update_manager_ui.py
@@ -13,6 +13,7 @@ import logging
 from pathlib import Path
 
 from PySide6.QtCore import Qt, Signal
+from PySide6.QtGui import QIcon
 from PySide6.QtWidgets import (
     QDialog,
     QHBoxLayout,
@@ -43,7 +44,7 @@ class CheckingDialog(QDialog):
         self.setWindowTitle("Checking for Updates")
         self.setModal(True)
         self.setMinimumWidth(300)
-        self.setWindowFlags(self.windowFlags() & ~Qt.WindowCloseButtonHint)
+        self.setWindowFlags(self.windowFlags() & ~Qt.WindowType.WindowCloseButtonHint)
         
         layout = QVBoxLayout()
         
@@ -165,7 +166,7 @@ class DownloadingDialog(QDialog):
         self.setWindowTitle("Downloading Update")
         self.setModal(True)
         self.setMinimumWidth(350)
-        self.setWindowFlags(self.windowFlags() & ~Qt.WindowCloseButtonHint)
+        self.setWindowFlags(self.windowFlags() & ~Qt.WindowType.WindowCloseButtonHint)
         
         layout = QVBoxLayout()
         
diff --git a/tests/unit/test_update_manager_ui.py b/tests/unit/test_update_manager_ui.py
index e1c5f6a..23f5d3e 100644
--- a/tests/unit/test_update_manager_ui.py
+++ b/tests/unit/test_update_manager_ui.py
@@ -44,7 +44,7 @@ class TestCheckingDialog:
         """Test dialog has no close button."""
         dialog = CheckingDialog()
         # WindowCloseButtonHint should be removed
-        assert not (dialog.windowFlags() & Qt.WindowCloseButtonHint)
+        assert not (dialog.windowFlags() & Qt.WindowType.WindowCloseButtonHint)
 
 
 class TestUpdateAvailableDialog:
@@ -141,7 +141,7 @@ class TestDownloadingDialog:
     def test_no_close_button(self, qapp):
         """Test dialog has no close button."""
         dialog = DownloadingDialog()
-        assert not (dialog.windowFlags() & Qt.WindowCloseButtonHint)
+        assert not (dialog.windowFlags() & Qt.WindowType.WindowCloseButtonHint)
 
 
 class TestInstallDialog: