diff --git a/include/common/tdataformat.h b/include/common/tdataformat.h
index c6ef6c513fd7cdec335ea8c0c9d8b51edaee85f6..d33f0af13c36aa1a2b2f66d91fb3d61f72302ff5 100644
--- a/include/common/tdataformat.h
+++ b/include/common/tdataformat.h
@@ -454,6 +454,7 @@ typedef struct {
 #define kvRowValLen(r) (kvRowLen(r) - TD_KV_ROW_HEAD_SIZE - sizeof(SColIdx) * kvRowNCols(r))
 #define kvRowTKey(r) (*(TKEY *)(kvRowValues(r)))
 #define kvRowKey(r) tdGetKey(kvRowTKey(r))
+#define kvRowKeys(r) POINTER_SHIFT(r, *(uint16_t *)POINTER_SHIFT(r, TD_KV_ROW_HEAD_SIZE + sizeof(int16_t)))
 #define kvRowDeleted(r) TKEY_IS_DELETED(kvRowTKey(r))
 
 SKVRow tdKVRowDup(SKVRow row);
@@ -654,6 +655,7 @@ static FORCE_INLINE char *memRowEnd(SMemRow row) {
 
 #define memRowTKey(r) (isDataRow(r) ? dataRowTKey(memRowDataBody(r)) : kvRowTKey(memRowKvBody(r)))
 #define memRowKey(r) (isDataRow(r) ? dataRowKey(memRowDataBody(r)) : kvRowKey(memRowKvBody(r)))
+#define memRowKeys(r) (isDataRow(r) ? dataRowTuple(memRowDataBody(r)) : kvRowKeys(memRowKvBody(r)))
 #define memRowSetTKey(r, k)                 \
   do {                                      \
     if (isDataRow(r)) {                     \
diff --git a/source/dnode/vnode/tsdb/src/tsdbMemTable.c b/source/dnode/vnode/tsdb/src/tsdbMemTable.c
index 1bfba55c86609d29b083f37f87669b365d1528e9..7a621c2a617451899e0f708395d76df491821d6b 100644
--- a/source/dnode/vnode/tsdb/src/tsdbMemTable.c
+++ b/source/dnode/vnode/tsdb/src/tsdbMemTable.c
@@ -318,7 +318,7 @@ static void tsdbFreeTbData(STbData *pTbData) {
   }
 }
 
-static char *tsdbGetTsTupleKey(const void *data) { return memRowKey((SMemRow)data); }
+static char *tsdbGetTsTupleKey(const void *data) { return memRowKeys((SMemRow)data); }
 
 static int tsdbTbDataComp(const void *arg1, const void *arg2) {
   STbData *pTbData1 = (STbData *)arg1;