test: improve wallet test