fix: PRT - fixing total stake amount taking only stake size into cons…

…ideration
lavanet · Oct 17, 2024 · b52ee8d · b52ee8d
1 parent 13593b4
commit b52ee8d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/protocol/statetracker/updaters/pairing_updater.go b/protocol/statetracker/updaters/pairing_updater.go
@@ -248,12 +248,13 @@ func (pu *PairingUpdater) filterPairingListByEndpoint(ctx context.Context, curre
 			pairingEndpoints[idx] = endp
 		}
 		lavasession.SortByGeolocations(pairingEndpoints, currentGeo)
+		totalStakeIncludingDelegation := sdk.Coin{Denom: provider.Stake.Denom, Amount: provider.Stake.Amount.Add(provider.DelegateTotal.Amount)}
 		pairing[uint64(providerIdx)] = lavasession.NewConsumerSessionWithProvider(
 			provider.Address,
 			pairingEndpoints,
 			maxCu,
 			epoch,
-			provider.Stake,
+			totalStakeIncludingDelegation,
 		)
 	}
 	if len(pairing) == 0 {