diff --git a/src/backend/distributed/operations/rebalancer_placement_separation.c b/src/backend/distributed/operations/rebalancer_placement_separation.c
index d5e51efb6..d7b1c1156 100644
--- a/src/backend/distributed/operations/rebalancer_placement_separation.c
+++ b/src/backend/distributed/operations/rebalancer_placement_separation.c
@@ -99,7 +99,6 @@ static int WorkerNodeListGetNodeWithGroupId(List *workerNodeList, int32 nodeGrou
 RebalancerPlacementSeparationContext *
 PrepareRebalancerPlacementSeparationContext(List *activeWorkerNodeList,
 											List *rebalancePlacementList,
-											WorkerNode *drainWorkerNode,
 											FmgrInfo *shardAllowedOnNodeUDF)
 {
 	HTAB *nodePlacementGroupHash =
@@ -115,7 +114,7 @@ PrepareRebalancerPlacementSeparationContext(List *activeWorkerNodeList,
 	rebalancePlacementList = SortList(rebalancePlacementList, CompareShardPlacements);
 
 	InitRebalancerPlacementSeparationContext(context, activeWorkerNodeList,
-											 rebalancePlacementList, drainWorkerNode);
+											 rebalancePlacementList);
 
 	TryAssignPlacementGroupsToNodeGroups(context,
 										 activeWorkerNodeList,
diff --git a/src/backend/distributed/operations/shard_rebalancer.c b/src/backend/distributed/operations/shard_rebalancer.c
index 5ba6bd23e..ab4f2d3c8 100644
--- a/src/backend/distributed/operations/shard_rebalancer.c
+++ b/src/backend/distributed/operations/shard_rebalancer.c
@@ -598,7 +598,6 @@ GetRebalanceSteps(RebalanceOptions *options)
 		PrepareRebalancerPlacementSeparationContext(
 			activeWorkerList,
 			FlattenNestedList(activeShardPlacementListList),
-			options->workerNode,
 			&context.shardAllowedOnNodeUDF);
 
 	return RebalancePlacementUpdates(activeWorkerList,
diff --git a/src/include/distributed/rebalancer_placement_separation.h b/src/include/distributed/rebalancer_placement_separation.h
index b00b3f87c..133265ca6 100644
--- a/src/include/distributed/rebalancer_placement_separation.h
+++ b/src/include/distributed/rebalancer_placement_separation.h
@@ -26,9 +26,6 @@ extern RebalancerPlacementSeparationContext * PrepareRebalancerPlacementSeparati
 	List
 	*
 	activeShardPlacementList,
-	WorkerNode
-	*
-	drainWorkerNode,
 	FmgrInfo
 	*
 	shardAllowedOnNodeUDF);