ydb-platform
diff --git a/‎ydb/core/tx/datashard/datashard__engine_host.h‎
Lines changed: 1 addition & 0 deletions b/‎ydb/core/tx/datashard/datashard__engine_host.h‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎ydb/core/tx/datashard/datashard_pipeline.cpp‎
Lines changed: 1 addition & 1 deletion b/‎ydb/core/tx/datashard/datashard_pipeline.cpp‎
Lines changed: 1 addition & 1 deletion
@@ -52,6 +52,7 @@ class TEngineBay : TNonCopyable {
 
     const NMiniKQL::IEngineFlat * GetEngine() const { return Engine.Get(); }
     NMiniKQL::IEngineFlat * GetEngine();
+    NMiniKQL::TEngineHost * GetEngineHost() { return EngineHost.Get(); }
     void SetLockTxId(ui64 lockTxId, ui32 lockNodeId);
     void SetUseLlvmRuntime(bool llvmRuntime) { EngineSettings->LlvmRuntime = llvmRuntime; }
 
 
@@ -1580,7 +1580,7 @@ TOperation::TPtr TPipeline::BuildOperation(NEvents::TDataEvents::TEvWrite::TPtr&
         return writeOp;
     }
 
-    writeOp->ExtractKeys();
+    writeTx->ExtractKeys(true);
 
     switch (rec.txmode()) {
         case NKikimrDataEvents::TEvWrite::MODE_PREPARE:
Original file line number	Diff line number	Diff line change
`@@ -1580,7 +1580,7 @@ TOperation::TPtr TPipeline::BuildOperation(NEvents::TDataEvents::TEvWrite::TPtr&`
`1580`	`1580`	`return writeOp;`
`1581`	`1581`	`}`
`1582`	`1582`
`1583`		`- writeOp->ExtractKeys();`
	`1583`	`+ writeTx->ExtractKeys(true);`
`1584`	`1584`
`1585`	`1585`	`switch (rec.txmode()) {`
`1586`	`1586`	`case NKikimrDataEvents::TEvWrite::MODE_PREPARE:`