Apply same optimization in ExistingEdgePartitionBuilder