diff --git a/core/trino-main/src/main/java/io/trino/cost/TaskCountEstimator.java b/core/trino-main/src/main/java/io/trino/cost/TaskCountEstimator.java index d4ccefe24ab6..1fa9312c7e6e 100644 --- a/core/trino-main/src/main/java/io/trino/cost/TaskCountEstimator.java +++ b/core/trino-main/src/main/java/io/trino/cost/TaskCountEstimator.java @@ -37,10 +37,11 @@ public class TaskCountEstimator public TaskCountEstimator(NodeSchedulerConfig nodeSchedulerConfig, InternalNodeManager nodeManager) { requireNonNull(nodeSchedulerConfig, "nodeSchedulerConfig is null"); + boolean schedulerIncludeCoordinator = nodeSchedulerConfig.isIncludeCoordinator(); requireNonNull(nodeManager, "nodeManager is null"); this.numberOfNodes = () -> { Set activeNodes = nodeManager.getAllNodes().getActiveNodes(); - if (nodeSchedulerConfig.isIncludeCoordinator()) { + if (schedulerIncludeCoordinator) { return activeNodes.size(); } return toIntExact(activeNodes.stream()