diff --git a/crates/iota-e2e-tests/tests/reconfiguration_tests.rs b/crates/iota-e2e-tests/tests/reconfiguration_tests.rs
index f7e1dfbc6a6..0c905ba0d0d 100644
--- a/crates/iota-e2e-tests/tests/reconfiguration_tests.rs
+++ b/crates/iota-e2e-tests/tests/reconfiguration_tests.rs
@@ -843,12 +843,11 @@ async fn safe_mode_reconfig_test() {
     assert_eq!(system_state.system_state_version, 1);
     assert_eq!(system_state.epoch, 0);
 
-    // Wait for regular epoch change to happen once. Migration from V1 to V2 should
-    // happen here.
+    // Wait for regular epoch change to happen once.
     let system_state = test_cluster.wait_for_epoch(Some(1)).await;
     assert!(!system_state.safe_mode());
     assert_eq!(system_state.epoch(), 1);
-    assert_eq!(system_state.system_state_version(), 2);
+    assert_eq!(system_state.system_state_version(), 1);
 
     let prev_epoch_start_timestamp = system_state.epoch_start_timestamp_ms();
 
@@ -877,7 +876,7 @@ async fn safe_mode_reconfig_test() {
     let system_state = test_cluster.wait_for_epoch(Some(3)).await;
     assert!(!system_state.safe_mode());
     assert_eq!(system_state.epoch(), 3);
-    assert_eq!(system_state.system_state_version(), 2);
+    assert_eq!(system_state.system_state_version(), 1);
 }
 
 async fn add_validator_candidate(