Fix tests for new NewWallet func signature.