From 2e94bda30d9f28435093c5737802cbf09727b0d0 Mon Sep 17 00:00:00 2001
From: Lane Rettig <lanerettig@gmail.com>
Date: Tue, 8 Oct 2024 16:25:30 -0700
Subject: [PATCH] Add pubkey back to wallet template

We may need this later, to pass into the VM template
---
 vm/templates/wallet/wallet.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/vm/templates/wallet/wallet.go b/vm/templates/wallet/wallet.go
index 27ba58503a..2422821326 100644
--- a/vm/templates/wallet/wallet.go
+++ b/vm/templates/wallet/wallet.go
@@ -8,13 +8,15 @@ import (
 
 // New returns Wallet instance with SpawnArguments.
 func New(args *SpawnArguments) *Wallet {
-	return &Wallet{}
+	return &Wallet{PublicKey: args.PublicKey}
 }
 
 //go:generate scalegen
 
 // Wallet is a single-key wallet.
-type Wallet struct{}
+type Wallet struct {
+	PublicKey core.PublicKey
+}
 
 // MaxSpend returns amount specified in the SpendArguments for Spend method.
 func (s *Wallet) MaxSpend(args any) (uint64, error) {