spool: Fix an error in exchange SimpleChildBuilder

apache · Oct 29, 2024 · 9be76e4 · 9be76e4
1 parent 257a83a
commit 9be76e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pinot-query-planner/src/test/java/org/apache/pinot/query/planner/logical/StagesTestBase.java b/pinot-query-planner/src/test/java/org/apache/pinot/query/planner/logical/StagesTestBase.java
@@ -119,7 +119,7 @@ public SimpleChildBuilder<JoinNode> join(
   public SimpleChildBuilder<MailboxReceiveNode> exchange(
       int nextStageId, SimpleChildBuilder<? extends PlanNode> childBuilder) {
     return (stageId, mySchema, myHints) -> {
-      PlanNode input = childBuilder.build(stageId);
+      PlanNode input = childBuilder.build(nextStageId);
       MailboxSendNode mailboxSendNode = new MailboxSendNode(nextStageId, null, List.of(input), stageId, null, null,
           null, false, null, false);
       MailboxSendNode old = _stageRoots.put(nextStageId, mailboxSendNode);