fix

subnet/subnet_test.go

@@ -7,9 +7,10 @@ import (
 	"avalanche-tooling-sdk-go/avalanche"
 	"context"
 	"fmt"
+	"testing"
+
 	"github.com/ava-labs/avalanchego/vms/secp256k1fx"
 	"github.com/ava-labs/avalanchego/wallet/subnet/primary"
-	"testing"
 )
 
 func TestSubnetDeploy(t *testing.T) {
@@ -27,7 +28,7 @@ func TestSubnetDeploy(t *testing.T) {
 	}
 	newSubnet := New(baseApp, &subnetParams)
 	ctx := context.Background()
-	wallet, _ := primary.MakeWallet(
+	wallet, _ := NewWallet(
 		ctx,
 		&primary.WalletConfig{
 			URI:              "",

subnet/wallet.go

@@ -4,6 +4,8 @@
 package subnet
 
 import (
+	"context"
+
 	"github.com/ava-labs/avalanchego/ids"
 	"github.com/ava-labs/avalanchego/utils/crypto/keychain"
 	"github.com/ava-labs/avalanchego/utils/set"
@@ -18,6 +20,17 @@ type Wallet struct {
 	options  []common.Option
 }
 
+func NewWallet(ctx context.Context, config *primary.WalletConfig) (Wallet, error) {
+	wallet, err := primary.MakeWallet(
+		ctx,
+		config,
+	)
+	return Wallet{
+		Wallet:   wallet,
+		keychain: config.AVAXKeychain,
+	}, err
+}
+
 // secure that a fee paying address (wallet's keychain) will receive the change,
 // and not a randomly selected auth key that may not be paying fees
 func (w *Wallet) SecureWalletIsChangeOwner() {