diff --git a/electrum/gui/qt/installwizard.py b/electrum/gui/qt/installwizard.py index c42ef7c9f..909aa7522 100644 --- a/electrum/gui/qt/installwizard.py +++ b/electrum/gui/qt/installwizard.py @@ -87,20 +87,20 @@ class CosignWidget(QWidget): def wizard_dialog(func): def func_wrapper(*args, **kwargs): run_next = kwargs['run_next'] - wizard = args[0] + wizard = args[0] # type: InstallWizard wizard.back_button.setText(_('Back') if wizard.can_go_back() else _('Cancel')) try: out = func(*args, **kwargs) + if type(out) is not tuple: + out = (out,) + run_next(*out) except GoBack: - wizard.go_back() if wizard.can_go_back() else wizard.close() - return - except UserCancelled: - return - #if out is None: - # out = () - if type(out) is not tuple: - out = (out,) - run_next(*out) + if wizard.can_go_back(): + wizard.go_back() + return + else: + wizard.close() + raise return func_wrapper