diff --git a/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp b/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp
index a02bbd976e25654028a2b98f93824bc939a22959..0aeac2c2a72ea928d64bb6a9a3e4f9b6a743e36d 100644
--- a/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp
+++ b/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp
@@ -200,7 +200,7 @@ bool OckMergeReader::CopyDataToVector(Vector *dstVector, uint32_t colIndex)
     auto *nullsAddress = (uint8_t *)dstVector->GetValueNulls();
     auto *valuesAddress = (uint8_t *)dstVector->GetValues();
     uint32_t *offsetsAddress = (uint32_t *)dstVector->GetValueOffsets();
-    dstVector->SetValueNulls(true);
+    dstVector->SetNullFlag(true);
     uint32_t totalSize = 0;
     uint32_t currentSize = 0;