simplify

multiversx · Aug 25, 2021 · 322637a · 322637a
1 parent db6bb03
commit 322637a
Showing 1 changed file with 8 additions and 14 deletions.
diff --git a/vm/systemSmartContracts/delegation.go b/vm/systemSmartContracts/delegation.go
@@ -1420,26 +1420,20 @@ func (d *delegation) addToActiveFund(
 	dStatus *DelegationContractStatus,
 	isNew bool,
 ) error {
-	if len(delegator.ActiveFund) == 0 {
-		var fundKey []byte
-		fundKey, err := d.createAndSaveNextKeyFund(callerAddr, delegateValue, active)
-		if err != nil {
-			return err
-		}
-
-		delegator.ActiveFund = fundKey
-		if isNew {
-			dStatus.NumUsers++
-		}
-
-		return nil
+	if len(delegator.ActiveFund) > 0 {
+		return d.addValueToFund(delegator.ActiveFund, delegateValue)
 	}
 
-	err := d.addValueToFund(delegator.ActiveFund, delegateValue)
+	fundKey, err := d.createAndSaveNextKeyFund(callerAddr, delegateValue, active)
 	if err != nil {
 		return err
 	}
 
+	delegator.ActiveFund = fundKey
+	if isNew {
+		dStatus.NumUsers++
+	}
+
 	return nil
 }