diff --git a/bitcoin_usb/device.py b/bitcoin_usb/device.py
index 5cfe891..1113af5 100644
--- a/bitcoin_usb/device.py
+++ b/bitcoin_usb/device.py
@@ -322,7 +322,10 @@ def __enter__(self):
                 logger.error(f"{filepath} returned {error=}")
 
             if not self.client.client.features.initialized:
-                if question_dialog(text=self.tr("Do you want to restore an existing seed onto the device?")):
+                if question_dialog(
+                    text=self.tr("Do you want to restore an existing seed onto the device?"),
+                    buttons=QMessageBox.StandardButton.No | QMessageBox.StandardButton.Yes,
+                ):
                     self.client.restore_device(label=self.initalization_label)
                 else:
                     self.client.setup_device(label=self.initalization_label)
@@ -335,7 +338,10 @@ def __enter__(self):
             self.noise_config = DialogNoiseConfig()
             self.client.noise_config = self.noise_config
             if not self.is_bitbox02_initialized(self.client):
-                if question_dialog(text=self.tr("Do you want to restore an existing seed onto the device?")):
+                if question_dialog(
+                    text=self.tr("Do you want to restore an existing seed onto the device?"),
+                    buttons=QMessageBox.StandardButton.No | QMessageBox.StandardButton.Yes,
+                ):
                     self.client.restore_device(label=self.initalization_label)
                 else:
                     self.client.setup_device(label=self.initalization_label)