Fix query balance response (cosmos#464)

notional-labs · Mar 22, 2021 · 1ff4ae1 · 1ff4ae1
1 parent 30b7ce5
commit 1ff4ae1
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/helpers/query.go b/helpers/query.go
@@ -38,12 +38,17 @@ func QueryBalance(chain *relayer.Chain, address string, showDenoms bool) (sdk.Co
 
 	var out sdk.Coins
 	for _, c := range coins {
-		for _, d := range dts.DenomTraces {
-			switch {
-			case c.Amount.Equal(sdk.NewInt(0)):
-			case c.Denom == d.IBCDenom():
-				out = append(out, sdk.NewCoin(d.IBCDenom(), c.Amount))
-			default:
+		if c.Amount.Equal(sdk.NewInt(0)) {
+			continue
+		}
+
+		for i, d := range dts.DenomTraces {
+			if c.Denom == d.IBCDenom() {
+				out = append(out, sdk.Coin{Denom: d.GetFullDenomPath(), Amount: c.Amount})
+				break
+			}
+
+			if i == len(dts.DenomTraces)-1 {
 				out = append(out, c)
 			}
 		}