Wallet fix