diff --git a/benchmark/local_infinity/sparse/sparse_benchmark_util.h b/benchmark/local_infinity/sparse/sparse_benchmark_util.h
index 87dec992f1..54af631300 100644
--- a/benchmark/local_infinity/sparse/sparse_benchmark_util.h
+++ b/benchmark/local_infinity/sparse/sparse_benchmark_util.h
@@ -24,6 +24,7 @@ import infinity_exception;
 import sparse_util;
 import compilation_config;
 import bmp_util;
+import local_file_handle;
 
 using namespace infinity;
 
diff --git a/src/storage/txn/txn.cpp b/src/storage/txn/txn.cpp
index e2e0e14552..65ed4cb0fd 100644
--- a/src/storage/txn/txn.cpp
+++ b/src/storage/txn/txn.cpp
@@ -583,7 +583,7 @@ void Txn::Rollback() {
 }
 
 void Txn::AddWalCmd(const SharedPtr<WalCmd> &cmd) { 
-    // std::lock_guard guard(txn_store_.mtx_);
+    std::lock_guard guard(txn_store_.mtx_); // TODO remove this line.
     auto state = txn_context_.GetTxnState();
     if (state != TxnState::kStarted) {
         auto begin_ts = BeginTS();