Improving test on Wallet