add if v2

cosmos · Jan 10, 2025 · af4b3d6 · af4b3d6
1 parent a7426a2
commit af4b3d6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/systemtests/upgrade_test.go b/tests/systemtests/upgrade_test.go
@@ -42,7 +42,12 @@ func TestChainUpgrade(t *testing.T) {
 		upgradeName         = "v050-to-v052" // must match UpgradeName in simapp/upgrades.go
 	)
 
-	systest.Sut.StartChain(t, fmt.Sprintf("--comet.halt-height=%d", upgradeHeight+1))
+	// if v2 is enabled, we need to start the chain with a different flag
+	if systest.IsV2() {
+		systest.Sut.StartChain(t, fmt.Sprintf("--comet.halt-height=%d", upgradeHeight+1))
+	} else {
+		systest.Sut.StartChain(t, fmt.Sprintf("--halt-height=%d", upgradeHeight+1))
+	}
 
 	cli := systest.NewCLIWrapper(t, systest.Sut, systest.Verbose)
 	govAddr := sdk.AccAddress(address.Module("gov")).String()