From 88803153c916098d65c04125a05f4848ab75c2c4 Mon Sep 17 00:00:00 2001
From: Wei-Ning Huang <w@dexon.org>
Date: Wed, 12 Dec 2018 10:08:01 +0800
Subject: core: governance interface should return correct DKG master public
 keys (#85)

---
 core/governance.go | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

(limited to 'core/governance.go')

diff --git a/core/governance.go b/core/governance.go
index 4b625c0fa..ea0036a0e 100644
--- a/core/governance.go
+++ b/core/governance.go
@@ -118,15 +118,7 @@ func (g *Governance) DKGComplaints(round uint64) []*dkgTypes.Complaint {
 
 func (g *Governance) DKGMasterPublicKeys(round uint64) []*dkgTypes.MasterPublicKey {
 	headHelper := g.GetHeadHelper()
-	var dkgMasterPKs []*dkgTypes.MasterPublicKey
-	for _, pk := range headHelper.DKGMasterPublicKeys(big.NewInt(int64(round))) {
-		x := new(dkgTypes.MasterPublicKey)
-		if err := rlp.DecodeBytes(pk, x); err != nil {
-			panic(err)
-		}
-		dkgMasterPKs = append(dkgMasterPKs, x)
-	}
-	return dkgMasterPKs
+	return headHelper.UniqueDKGMasterPublicKeys(big.NewInt(int64(round)))
 }
 
 func (g *Governance) IsDKGFinal(round uint64) bool {
-- 
cgit v1.2.3