added method to wallet mock