fix: tidy up collateral balances

vegaprotocol · Dec 20, 2023 · 3124fb4 · 3124fb4
1 parent 60466bc
commit 3124fb4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/core/collateral/engine.go b/core/collateral/engine.go
@@ -3696,6 +3696,7 @@ func (e *Engine) RemoveDistressed(ctx context.Context, parties []events.MarketPo
 			if err := e.UpdateBalance(ctx, bondAcc.ID, bondAcc.Balance.SetUint64(0)); err != nil {
 				return nil, err
 			}
+			// get an updated copy of the margin account after being topped up from bond account
 			marginAcc, _ = e.GetAccountByID(e.accountID(marketID, party.Party(), asset, types.AccountTypeMargin))
 		}
 		// take whatever is left on the general account, and move to margin balance
@@ -3717,6 +3718,7 @@ func (e *Engine) RemoveDistressed(ctx context.Context, parties []events.MarketPo
 			if err := e.UpdateBalance(ctx, genAcc.ID, genAcc.Balance.SetUint64(0)); err != nil {
 				return nil, err
 			}
+			// get an updated copy of the margin account after being topped up from general account
 			marginAcc, _ = e.GetAccountByID(e.accountID(marketID, party.Party(), asset, types.AccountTypeMargin))
 		}
 		// move monies from the margin account (balance is general, bond, and margin combined now)