From 5baa5feb114b21147fc36d3df2916455a251da90 Mon Sep 17 00:00:00 2001
From: Neil Booth <kyuupichan@gmail.com>
Date: Sun, 10 Jan 2016 18:41:20 +0900
Subject: [PATCH] Put restore_wallet_class back in the derived classes

---
 lib/wallet.py            | 1 -
 plugins/ledger/ledger.py | 1 +
 plugins/trezor/plugin.py | 2 ++
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/lib/wallet.py b/lib/wallet.py
index 12c7feedf..a7929c1b8 100644
--- a/lib/wallet.py
+++ b/lib/wallet.py
@@ -1746,7 +1746,6 @@ class BIP32_HD_Wallet(BIP32_Wallet):
 class BIP44_Wallet(BIP32_HD_Wallet):
     root_derivation = "m/44'/0'"
     wallet_type = 'bip44'
-    restore_wallet_class = BIP44_Wallet
 
     def can_sign_xpubkey(self, x_pubkey):
         xpub, sequence = BIP32_Account.parse_xpubkey(x_pubkey)
diff --git a/plugins/ledger/ledger.py b/plugins/ledger/ledger.py
index 4b5f0170a..e0ce5957a 100644
--- a/plugins/ledger/ledger.py
+++ b/plugins/ledger/ledger.py
@@ -32,6 +32,7 @@ except ImportError:
 class BTChipWallet(BIP44_Wallet):
     wallet_type = 'btchip'
     device = 'Ledger'
+    restore_wallet_class = BIP44_Wallet
 
     def __init__(self, storage):
         BIP44_Wallet.__init__(self, storage)
diff --git a/plugins/trezor/plugin.py b/plugins/trezor/plugin.py
index 7e7aa26a4..cbbbbae40 100644
--- a/plugins/trezor/plugin.py
+++ b/plugins/trezor/plugin.py
@@ -32,6 +32,8 @@ class TrezorCompatibleWallet(BIP44_Wallet):
     #   - DEVICE_IDS
     #   - wallet_type
 
+    restore_wallet_class = BIP44_Wallet
+
     def __init__(self, storage):
         BIP44_Wallet.__init__(self, storage)
         # After timeout seconds we clear the device session