address the comment

Signed-off-by: Ryan Leung <rleungx@gmail.com>
tikv · Sep 10, 2021 · f00638e · f00638e
1 parent 2c5d9ba
commit f00638e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/server/schedule/operator/step.go b/server/schedule/operator/step.go
@@ -262,6 +262,7 @@ func (rp RemovePeer) Influence(opInfluence OpInfluence, region *core.RegionInfo)
 	from.RegionSize -= regionSize
 	from.RegionCount--
 	if rp.IsDownStore {
+		from.AdjustStepCost(storelimit.RemovePeer, storelimit.SmallRegionThreshold)
 		return
 	}
 	from.AdjustStepCost(storelimit.RemovePeer, regionSize)