diff --git a/lbrynet/core/Wallet.py b/lbrynet/core/Wallet.py index 59f0b7574..38b6aabcb 100644 --- a/lbrynet/core/Wallet.py +++ b/lbrynet/core/Wallet.py @@ -15,7 +15,7 @@ from lbryum import wallet as lbryum_wallet from lbryum.network import Network from lbryum.simple_config import SimpleConfig from lbryum.constants import COIN -from lbryum.commands import known_commands, Commands +from lbryum.commands import Commands from lbryschema.uri import parse_lbry_uri from lbryschema.claim import ClaimDict @@ -1305,7 +1305,7 @@ class LBRYumWallet(Wallet): # only makes a lbrum server query, use _run_cmd_as_defer_to_thread() def _run_cmd_as_defer_succeed(self, command_name, *args, **kwargs): cmd_runner = self._get_cmd_runner() - cmd = known_commands[command_name] + cmd = Commands.known_commands[command_name] func = getattr(cmd_runner, cmd.name) return defer.succeed(func(*args, **kwargs)) @@ -1315,7 +1315,7 @@ class LBRYumWallet(Wallet): # otherwise the application will hang, waiting for threads to complete def _run_cmd_as_defer_to_thread(self, command_name, *args, **kwargs): cmd_runner = self._get_cmd_runner() - cmd = known_commands[command_name] + cmd = Commands.known_commands[command_name] func = getattr(cmd_runner, cmd.name) return threads.deferToThread(func, *args, **kwargs)