diff --git a/gui/qt/installwizard.py b/gui/qt/installwizard.py index b5e3e497e..250552ef1 100644 --- a/gui/qt/installwizard.py +++ b/gui/qt/installwizard.py @@ -538,6 +538,7 @@ class InstallWizard(QDialog): wallet.create_main_account(password) else: + self.storage.put('wallet_type', t) wallet = run_hook('installwizard_restore', self, self.storage) if not wallet: return diff --git a/plugins/trezor.py b/plugins/trezor.py index 12e8c8e55..79bb85e0d 100644 --- a/plugins/trezor.py +++ b/plugins/trezor.py @@ -81,6 +81,8 @@ class Plugin(BasePlugin): wallet_types.append(('trezor', _("Trezor wallet"), TrezorWallet)) def installwizard_restore(self, wizard, storage): + if storage.get('wallet_type') != 'trezor': + return wallet = TrezorWallet(storage) try: wallet.create_main_account(None)