Expand apply for lambdas inside views

1 files changed, 4 insertions, 1 deletions

diff --git a/ydb/library/yql/core/yql_opt_rewrite_io.cpp b/ydb/library/yql/core/yql_opt_rewrite_io.cpp
index ea96f34b680..40fc5aa22d7 100644
--- a/ydb/library/yql/core/yql_opt_rewrite_io.cpp
+++ b/ydb/library/yql/core/yql_opt_rewrite_io.cpp
@@ -34,7 +34,10 @@ IGraphTransformer::TStatus RewriteIO(const TExprNode::TPtr& input, TExprNode::TP
         return ret;
     }
 
-    if (!ctx.Step.IsDone(TExprStep::DiscoveryIO)) {
+    if (
+        !ctx.Step.IsDone(TExprStep::DiscoveryIO) ||
+        !ctx.Step.IsDone(TExprStep::ExpandApplyForLambdas)
+    ) {
         return IGraphTransformer::TStatus(IGraphTransformer::TStatus::Repeat, true);
     }