diff --git a/run_electrum b/run_electrum
index 93aab3800..b5d5af00d 100755
--- a/run_electrum
+++ b/run_electrum
@@ -68,6 +68,7 @@ if not is_android:
 from electrum import bitcoin, util
 from electrum import constants
 from electrum import SimpleConfig, Network
+from electrum.wallet import Wallet, Imported_Wallet
 from electrum import bitcoin, util, constants
 from electrum.storage import WalletStorage, get_derivation_used_for_hw_device_encryption
 from electrum.util import print_msg, print_stderr, json_encode, json_decode, UserCancelled