fix --jail-whitelist?

scrtlabs · Aug 30, 2020 · a245976 · a245976
1 parent 6338180
commit a245976
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/app.go b/app.go
@@ -574,9 +574,12 @@ func (app *SecretNetworkApp) prepForZeroHeightGenesis(ctx sdk.Context, jailWhite
 		validator.UnbondingHeight = 0
 		if applyWhiteList && !whiteListMap[addr.String()] {
 			validator.Jailed = true
+			app.stakingKeeper.SetValidator(ctx, validator)
+			app.stakingKeeper.DeleteValidatorByPowerIndex(ctx, validator)
+		} else {
+			app.stakingKeeper.SetValidator(ctx, validator)
 		}
 
-		app.stakingKeeper.SetValidator(ctx, validator)
 		counter++
 	}