diff --git a/bazel/repositories.bzl b/bazel/repositories.bzl
index 1aa54f9..a7c9daa 100644
--- a/bazel/repositories.bzl
+++ b/bazel/repositories.bzl
@@ -51,10 +51,10 @@ def _yacl():
         http_archive,
         name = "yacl",
         urls = [
-            "https://github.com/secretflow/yacl/archive/refs/tags/0.4.5b1.tar.gz",
+            "https://github.com/secretflow/yacl/archive/refs/tags/0.4.5b3_nightly_20240722.tar.gz",
         ],
-        strip_prefix = "yacl-0.4.5b1",
-        sha256 = "28064053b9add0db8e1e8e648421a0579f1d3e7ee8a4bbd7bd5959cb59598088",
+        strip_prefix = "yacl-0.4.5b3_nightly_20240722",
+        sha256 = "ccca599e6ded6089c5afbb87c8f5e09383195af256caacd50089f0c7443e8604",
     )
 
 def _bazel_platform():
diff --git a/psi/kkrt/kkrt_psi.h b/psi/kkrt/kkrt_psi.h
index 3201e8f..7ad665a 100644
--- a/psi/kkrt/kkrt_psi.h
+++ b/psi/kkrt/kkrt_psi.h
@@ -21,7 +21,7 @@
 #include <unordered_map>
 #include <vector>
 
-#include "yacl/kernel/algorithms/ot_store.h"
+#include "yacl/kernel/type/ot_store.h"
 #include "yacl/link/link.h"
 
 //
diff --git a/psi/kkrt/receiver.h b/psi/kkrt/receiver.h
index aa7d694..98b44f0 100644
--- a/psi/kkrt/receiver.h
+++ b/psi/kkrt/receiver.h
@@ -13,7 +13,7 @@
 // limitations under the License.
 #pragma once
 
-#include "yacl/kernel/algorithms/ot_store.h"
+#include "yacl/kernel/type/ot_store.h"
 
 #include "psi/interface.h"
 #include "psi/utils/hash_bucket_cache.h"
diff --git a/psi/kkrt/sender.h b/psi/kkrt/sender.h
index d9b3cee..b6b4363 100644
--- a/psi/kkrt/sender.h
+++ b/psi/kkrt/sender.h
@@ -13,7 +13,7 @@
 // limitations under the License.
 #pragma once
 
-#include "yacl/kernel/algorithms/ot_store.h"
+#include "yacl/kernel/type/ot_store.h"
 
 #include "psi/interface.h"
 #include "psi/utils/hash_bucket_cache.h"