diff --git a/src/sf_binlog_writer.c b/src/sf_binlog_writer.c
index c9319ed..fa5a2ba 100644
--- a/src/sf_binlog_writer.c
+++ b/src/sf_binlog_writer.c
@@ -38,6 +38,9 @@ static inline void binlog_writer_set_next_version(SFBinlogWriterInfo *writer,
         const uint64_t next_version)
 {
     writer->version_ctx.next = next_version;
+    if (writer->fw.flags & SF_FILE_WRITER_FLAGS_WANT_DONE_VERSION) {
+        writer->fw.last_versions.done = next_version - 1;
+    }
 }
 
 #define deal_binlog_one_record(wb) \
diff --git a/src/sf_serializer.h b/src/sf_serializer.h
index 40739ca..aeff9eb 100644
--- a/src/sf_serializer.h
+++ b/src/sf_serializer.h
@@ -19,6 +19,7 @@
 #define _SF_SERIALIZER_H
 
 #include "fastcommon/common_define.h"
+#include "fastcommon/shared_func.h"
 #include "fastcommon/fast_buffer.h"
 #include "fastcommon/hash.h"