set alloc root

0chain · Jan 11, 2025 · f72c706 · f72c706
1 parent b336a0b
commit f72c706
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/zboxcore/sdk/rollback.go b/zboxcore/sdk/rollback.go
@@ -516,6 +516,7 @@ func (a *Allocation) checkStatusV2(markerChan chan *RollbackBlobber, blobStatus
 	}
 
 	if len(latestVersionMap[allVersionConensus]) >= a.DataShards {
+		a.allocationRoot = allVersionConensus
 		return Repair, blobStatus, nil
 	}
 	l.Logger.Info("Rolling back to previous version")
@@ -545,6 +546,7 @@ func (a *Allocation) checkStatusV2(markerChan chan *RollbackBlobber, blobStatus
 	if successCnt < int32(consensusThresh) {
 		return Broken, blobStatus, common.NewError("rollback_failed", "Rollback failed")
 	}
+	a.allocationRoot = allVersionConensus
 	if successCnt == int32(fullConsensus) {
 		return Repair, blobStatus, nil
 	}