Enable get result from CnMerge.

1 files changed, 3 insertions, 3 deletions

diff --git a/ydb/library/yql/providers/dq/provider/yql_dq_datasource.cpp b/ydb/library/yql/providers/dq/provider/yql_dq_datasource.cpp
index b617b7da7c0..c7c26aad026 100644
--- a/ydb/library/yql/providers/dq/provider/yql_dq_datasource.cpp
+++ b/ydb/library/yql/providers/dq/provider/yql_dq_datasource.cpp
@@ -126,8 +126,8 @@ public:
             return node;
         }
 
-        if (!TDqCnUnionAll::Match(node.Get())) {
-            ctx.AddError(TIssue(node->Pos(ctx), "Last connection must be union all"));
+        if (!(TDqCnUnionAll::Match(node.Get()) || TDqCnMerge::Match(node.Get()))) {
+            ctx.AddError(TIssue(node->Pos(ctx), "Last connection must be union all or merge"));
             return {};
         }
 
@@ -165,7 +165,7 @@ public:
     }
 
     bool CanPullResult(const TExprNode& node, TSyncMap& syncList, bool& canRef) override {
-        if (!TDqCnUnionAll::Match(&node)) {
+        if (!(TDqCnUnionAll::Match(&node) || TDqCnMerge::Match(&node))) {
             return false;
         }