1 files changed, 5 insertions, 2 deletions
diff --git a/src/qt/bitcoin.cpp b/src/qt/bitcoin.cpp
index c31be1b606..397af5fd38 100644
--- a/src/qt/bitcoin.cpp
+++ b/src/qt/bitcoin.cpp
@@ -3,6 +3,7 @@
  */
 #include "bitcoingui.h"
 #include "clientmodel.h"
+#include "walletmodel.h"
 
 #include "headers.h"
 #include "init.h"
@@ -113,9 +114,11 @@ int main(int argc, char *argv[])
         if(AppInit2(argc, argv))
         {
             BitcoinGUI window;
-            ClientModel model(pwalletMain);
+            ClientModel clientModel(pwalletMain);
+            WalletModel walletModel(pwalletMain);
             guiref = &window;
-            window.setModel(&model);
+            window.setClientModel(&clientModel);
+            window.setWalletModel(&walletModel);
 
             window.show();