fixing revert invariant

dymensionxyz · Mar 18, 2024 · 0b17218 · 0b17218
1 parent 3e35a87
commit 0b17218
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/x/delayedack/keeper/invariants.go b/x/delayedack/keeper/invariants.go
@@ -21,6 +21,10 @@ func AllInvariants(k Keeper) sdk.Invariant {
 		if stop {
 			return res, stop
 		}
+		res, stop = RollappRevertedPackets(k)(ctx)
+		if stop {
+			return res, stop
+		}
 		return "", false
 	}
 }
@@ -83,7 +87,7 @@ func RollappRevertedPackets(k Keeper) sdk.Invariant {
 			if packet.ProofHeight > latestFinalizedHeight {
 				stateInfoIndex := latestFinalizedStateIndex.Index + 1
 				for {
-					stateInfoToCheck, found := k.rollappKeeper.GetStateInfo(ctx, packet.RollappId, latestFinalizedStateIndex.Index)
+					stateInfoToCheck, found := k.rollappKeeper.GetStateInfo(ctx, packet.RollappId, stateInfoIndex)
 					if found {
 						if stateInfoToCheck.Status == commontypes.Status_REVERTED {
 							if stateInfoToCheck.StartHeight >= packet.ProofHeight && stateInfoToCheck.StartHeight+stateInfoToCheck.NumBlocks < packet.ProofHeight {