From bc2893404317a2ac82a1dff28c156057e18ff657 Mon Sep 17 00:00:00 2001
From: Nguyen Sy Thanh Son <thanhson1085@gmail.com>
Date: Wed, 8 Aug 2018 04:30:46 +0000
Subject: [PATCH] fix test reward

---
 contracts/utils.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/contracts/utils.go b/contracts/utils.go
index cd357519a4..b239e22e8d 100644
--- a/contracts/utils.go
+++ b/contracts/utils.go
@@ -216,12 +216,18 @@ func GetRewardBalancesRate(masterAddr common.Address, totalReward *big.Int, vali
 				log.Error("Fail to get vote capacity", "error", err)
 				return nil, err
 			}
+
 			totalCap.Add(totalCap, voterCap)
 			voterCaps[voteAddr] = voterCap
 		}
 		for addr, voteCap := range voterCaps {
-			balances[addr] = new(big.Int).Mul(totalVoterReward, voteCap)
-			balances[addr] = new(big.Int).Div(balances[addr], totalCap)
+			rcap := new(big.Int).Mul(totalVoterReward, voteCap)
+			rcap = new(big.Int).Div(rcap, totalCap)
+			if balances[addr] != nil {
+				balances[addr].Add(balances[addr], rcap)
+			} else {
+				balances[addr] = rcap
+			}
 		}
 	}