From bf50bc45efea162a01583ea47a20791494909dc5 Mon Sep 17 00:00:00 2001
From: ThomasV <thomasv@gitorious>
Date: Thu, 11 Jun 2015 02:13:12 +0200
Subject: [PATCH] fix plugin load_wallet hook

---
 plugins/btchipwallet.py | 2 ++
 plugins/trustedcoin.py  | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/plugins/btchipwallet.py b/plugins/btchipwallet.py
index e51c9e40..e1aec9dc 100644
--- a/plugins/btchipwallet.py
+++ b/plugins/btchipwallet.py
@@ -74,6 +74,8 @@ class Plugin(BasePlugin):
 
     @hook
     def load_wallet(self, wallet, window):
+        self.wallet = wallet
+        self.window = window
         if self.btchip_is_connected():
             if not self.wallet.check_proper_device():
                 QMessageBox.information(self.window, _('Error'), _("This wallet does not match your BTChip device"), _('OK'))
diff --git a/plugins/trustedcoin.py b/plugins/trustedcoin.py
index 95f12a19..810c32af 100644
--- a/plugins/trustedcoin.py
+++ b/plugins/trustedcoin.py
@@ -323,6 +323,8 @@ class Plugin(BasePlugin):
 
     @hook
     def load_wallet(self, wallet, window):
+        self.wallet = wallet
+        self.window = window
         self.trustedcoin_button = StatusBarButton( QIcon(":icons/trustedcoin.png"), _("Network"), self.settings_dialog)
         self.window.statusBar().addPermanentWidget(self.trustedcoin_button)
         self.xpub = self.wallet.master_public_keys.get('x1/')