fix: check if discount nil

elys-network · Nov 25, 2023 · 221379a · 221379a
1 parent 5a98d93
commit 221379a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/x/amm/keeper/apply_discount.go b/x/amm/keeper/apply_discount.go
@@ -8,6 +8,12 @@ import (
 
 // ApplyDiscount applies discount to swap fee if applicable
 func (k Keeper) ApplyDiscount(ctx sdk.Context, swapFee sdk.Dec, discount sdk.Dec, sender string) (sdk.Dec, sdk.Dec, error) {
+	// if discount is nil, return swap fee and zero discount
+	if discount.IsNil() {
+		return swapFee, sdk.ZeroDec(), nil
+	}
+
+	// if discount is zero, return swap fee and zero discount
 	if discount.IsZero() {
 		return swapFee, sdk.ZeroDec(), nil
 	}