diff --git a/Cargo.lock b/Cargo.lock
index 5e71b5c64fe8..c54444c6f60d 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -7284,7 +7284,7 @@ dependencies = [
 [[package]]
 name = "raft-engine"
 version = "0.4.0"
-source = "git+https://github.com/tikv/raft-engine.git?rev=571462e36621407b9920465a1a15b8b01b929a7f#571462e36621407b9920465a1a15b8b01b929a7f"
+source = "git+https://github.com/tikv/raft-engine.git?rev=22dfb426cd994602b57725ef080287d3e53db479#22dfb426cd994602b57725ef080287d3e53db479"
 dependencies = [
  "byteorder",
  "crc32fast",
diff --git a/Cargo.toml b/Cargo.toml
index 647df95383f9..be1cf2d8bdd2 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -152,8 +152,7 @@ object-store = { path = "src/object-store" }
 partition = { path = "src/partition" }
 promql = { path = "src/promql" }
 query = { path = "src/query" }
-# TODO(weny): waits for https://github.com/tikv/raft-engine/pull/335
-raft-engine = { git = "https://github.com/tikv/raft-engine.git", rev = "571462e36621407b9920465a1a15b8b01b929a7f" }
+raft-engine = { git = "https://github.com/tikv/raft-engine.git", rev = "22dfb426cd994602b57725ef080287d3e53db479" }
 script = { path = "src/script" }
 servers = { path = "src/servers" }
 session = { path = "src/session" }