diff --git a/src/graph/optimizer/rule/PushFilterDownProjectRule.cpp b/src/graph/optimizer/rule/PushFilterDownProjectRule.cpp index 69ea357e248..d14fe98ee78 100644 --- a/src/graph/optimizer/rule/PushFilterDownProjectRule.cpp +++ b/src/graph/optimizer/rule/PushFilterDownProjectRule.cpp @@ -110,6 +110,8 @@ StatusOr PushFilterDownProjectRule::transform( auto* newBelowFilterNode = graph::Filter::make( octx->qctx(), const_cast(oldProjNode->dep()), newFilterPicked); newBelowFilterNode->setInputVar(oldProjNode->inputVar()); + // Filter should keep column names + newBelowFilterNode->setColNames(oldProjNode->inputVars()[0]->colNames); auto newBelowFilterGroup = OptGroup::create(octx); auto newFilterGroupNode = newBelowFilterGroup->makeGroupNode(newBelowFilterNode); for (auto dep : projGroupNode->dependencies()) {