fix metho name in update firmware ref fibercrypto#165

simelo · Feb 5, 2020 · 4c0e96d · 4c0e96d
1 parent 06acbc9
commit 4c0e96d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/cli/firmware_update.go b/src/cli/firmware_update.go
@@ -4,6 +4,7 @@ import (
 	"crypto/sha256"
 	"fmt"
 	"github.com/fibercrypto/skywallet-go/src/skywallet"
+	messages "github.com/fibercrypto/skywallet-protob/go"
 	"io/ioutil"
 
 	gcli "github.com/urfave/cli"
@@ -34,7 +35,8 @@ func firmwareUpdate() gcli.Command {
 			if err != nil {
 				return
 			}
-			err = sq.FirmwareUpload(firmware, sha256.Sum256(firmware[0x100:]))
+			msg, err := sq.UploadFirmware(firmware, sha256.Sum256(firmware[0x100:]))
+			handleFinalResponse(msg, err, "unable to update firmware", messages.MessageType_MessageType_Success)
 			if err != nil {
 				log.Error(err)
 				return