Have close_wallet hook take a wallet.

plugins/ledger/ledger.py

@@ -434,13 +434,11 @@ class LedgerPlugin(BasePlugin):
         return True
 
     @hook
-    def close_wallet(self):
-        pass
+    def close_wallet(self, wallet):
+        self.client = None
 
     @hook
     def installwizard_load_wallet(self, wallet, window):
-        if type(wallet) != BTChipWallet:
-            return
         self.load_wallet(wallet, window)
 
     @hook

plugins/trezor/plugin.py

@@ -193,9 +193,9 @@ class TrezorCompatiblePlugin(BasePlugin):
         return self.get_client().atleast_version(major, minor, patch)
 
     @hook
-    def close_wallet(self):
-        self.print_error("clear session")
+    def close_wallet(self, wallet):
         if self.client:
+            self.print_error("clear session")
             self.client.clear_session()
             self.client.transport.close()
             self.client = None

plugins/trezor/qt_generic.py

@@ -124,9 +124,6 @@ class QtPlugin(TrezorPlugin):
 
     @hook
     def installwizard_load_wallet(self, wallet, window):
-        if type(wallet) != self.wallet_class:
-            return
-        wallet.plugin = self
         self.load_wallet(wallet, window)
 
     @hook