fix wallet test