Update test_wallets.py