diff --git a/wallet/wallet_test.go b/wallet/wallet_test.go
index 9dbdca1f4..d5e9f8cee 100644
--- a/wallet/wallet_test.go
+++ b/wallet/wallet_test.go
@@ -578,10 +578,11 @@ func TestSignMessage(t *testing.T) {
 	msg := "pactus"
 	expectedSig := "923d67a8624cbb7972b29328e15ec76cc846076ccf00a9e94d991c677846f334ae4ba4551396fbcd6d1cab7593baf3b7"
 
-	prv, _ := bls.PrivateKeyFromString(
-		"SECRET1PDRWTLP5PX0FAHDX39GXZJP7FKZFALML0D5U9TT9KVQHDUC99CMGQQJVK67")
+	prv, err := bls.PrivateKeyFromString("SECRET1PDRWTLP5PX0FAHDX39GXZJP7FKZFALML0D5U9TT9KVQHDUC99CMGQQJVK67")
 
-	err := td.wallet.ImportPrivateKey("", prv)
+	assert.NoError(t, err)
+
+	err = td.wallet.ImportPrivateKey(td.password, prv)
 	assert.NoError(t, err)
 
 	sig, err := td.wallet.SignMessage(msg, td.password, td.wallet.AllAccountAddresses()[0].Address)
diff --git a/www/grpc/buf/grpc-gateway.config.yaml b/www/grpc/buf/grpc-gateway.config.yaml
index 8133eb71f..a3fe730d2 100644
--- a/www/grpc/buf/grpc-gateway.config.yaml
+++ b/www/grpc/buf/grpc-gateway.config.yaml
@@ -96,4 +96,5 @@ http:
       get: "/pactus/Utils/Sign_Message_With_PrivateKey"
 
     - selector: pactus.Utils.VerifyMessage
-      get: "/pactus/Utils/VerifyMessage"
\ No newline at end of file
+      get: "/pactus/Utils/VerifyMessage"
+      
\ No newline at end of file