fix: add weight require (#10)

MassDotMoney · Feb 16, 2022 · e82319e · e82319e
1 parent 720b450
commit e82319e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/contracts/FeeSplitter.sol b/contracts/FeeSplitter.sol
@@ -130,6 +130,7 @@ contract FeeSplitter is Ownable, ReentrancyGuard {
     /// @param _accountIndex Account to change the weight of
     /// @param _weight The new weight
     function updateShareholder(uint256 _accountIndex, uint96 _weight) external onlyOwner {
+        require(_weight != 0, "FS: INVALID_WEIGHT");
         require(_accountIndex < shareholders.length, "FS: INVALID_ACCOUNT_INDEX");
         totalWeights = totalWeights + _weight - shareholders[_accountIndex].weight;
         require(totalWeights != 0, "FS: TOTAL_WEIGHTS_ZERO");