Minor change

Ethernal-Tech · Feb 19, 2024 · 340c128 · 340c128
1 parent e1fa4c1
commit 340c128
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/command/secrets/init/params_test.go b/command/secrets/init/params_test.go
@@ -8,9 +8,9 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	"github.com/umbracle/ethgo/wallet"
 
 	"github.com/0xPolygon/polygon-edge/bls"
+	"github.com/0xPolygon/polygon-edge/crypto"
 	"github.com/0xPolygon/polygon-edge/secrets/helper"
 )
 
@@ -94,7 +94,8 @@ func Test_getResult(t *testing.T) {
 	// Test public key serialization
 	privKey, err := hex.DecodeString(sir.PrivateKey)
 	require.NoError(t, err)
-	k, err := wallet.NewWalletFromPrivKey(privKey)
+
+	k, err := crypto.NewECDSAKeyFromRawPrivECDSA(privKey)
 	require.NoError(t, err)
 
 	pubKey := k.Address().String()

diff --git a/consensus/polybft/wallet/account.go b/consensus/polybft/wallet/account.go
@@ -5,8 +5,6 @@ import (
 	"encoding/hex"
 	"fmt"
 
-	"github.com/umbracle/ethgo/wallet"
-
 	"github.com/0xPolygon/polygon-edge/bls"
 	"github.com/0xPolygon/polygon-edge/crypto"
 	"github.com/0xPolygon/polygon-edge/secrets"
@@ -117,7 +115,7 @@ func (a *Account) GetEcdsaPrivateKey() (*ecdsa.PrivateKey, error) {
 		return nil, err
 	}
 
-	return wallet.ParsePrivateKey(ecdsaRaw)
+	return crypto.ParseECDSAPrivateKey(ecdsaRaw)
 }
 
 func (a Account) Address() types.Address {