diff --git a/frontend/rust-lib/flowy-document/Cargo.toml b/frontend/rust-lib/flowy-document/Cargo.toml
index 813d87d589..1b2cf3c593 100644
--- a/frontend/rust-lib/flowy-document/Cargo.toml
+++ b/frontend/rust-lib/flowy-document/Cargo.toml
@@ -45,6 +45,7 @@ getrandom = { version = "0.2", features = ["js"] }
 tempfile = "3.4.0"
 tracing-subscriber = { version = "0.3.3", features = ["env-filter"] }
 collab-integrate = { workspace = true }
+tokio = { workspace = true, features = ["rt", "rt-multi-thread"] }
 
 [build-dependencies]
 flowy-codegen.workspace = true
diff --git a/frontend/rust-lib/flowy-document/tests/document/document_redo_undo_test.rs b/frontend/rust-lib/flowy-document/tests/document/document_redo_undo_test.rs
index 54be6a2d52..fc8268fd71 100644
--- a/frontend/rust-lib/flowy-document/tests/document/document_redo_undo_test.rs
+++ b/frontend/rust-lib/flowy-document/tests/document/document_redo_undo_test.rs
@@ -22,6 +22,7 @@ async fn undo_redo_test() {
     .await;
 
   // open a document
+  test.open_document(&doc_id).await.unwrap();
   let document = test.get_document(&doc_id).await.unwrap();
   let document = document.lock();
   let page_block = document.get_block(&data.page_id).unwrap();
diff --git a/frontend/rust-lib/flowy-document/tests/document/document_test.rs b/frontend/rust-lib/flowy-document/tests/document/document_test.rs
index 1296bc3c7c..754cba2ef0 100644
--- a/frontend/rust-lib/flowy-document/tests/document/document_test.rs
+++ b/frontend/rust-lib/flowy-document/tests/document/document_test.rs
@@ -20,6 +20,7 @@ async fn restore_document() {
     .unwrap();
   let data_a = test.get_document_data(&doc_id).await.unwrap();
   assert_eq!(data_a, data);
+  test.open_document(&doc_id).await.unwrap();
 
   let data_b = test
     .get_document(&doc_id)
@@ -59,6 +60,7 @@ async fn document_apply_insert_action() {
   _ = test.create_document(uid, &doc_id, Some(data.clone())).await;
 
   // open a document
+  test.open_document(&doc_id).await.unwrap();
   let document = test.get_document(&doc_id).await.unwrap();
   let page_block = document.lock().get_block(&data.page_id).unwrap();
 
@@ -112,6 +114,7 @@ async fn document_apply_update_page_action() {
   _ = test.create_document(uid, &doc_id, Some(data.clone())).await;
 
   // open a document
+  test.open_document(&doc_id).await.unwrap();
   let document = test.get_document(&doc_id).await.unwrap();
   let page_block = document.lock().get_block(&data.page_id).unwrap();
 
@@ -155,6 +158,7 @@ async fn document_apply_update_action() {
   _ = test.create_document(uid, &doc_id, Some(data.clone())).await;
 
   // open a document
+  test.open_document(&doc_id).await.unwrap();
   let document = test.get_document(&doc_id).await.unwrap();
   let page_block = document.lock().get_block(&data.page_id).unwrap();
 
diff --git a/frontend/rust-lib/flowy-document/tests/document/util.rs b/frontend/rust-lib/flowy-document/tests/document/util.rs
index 4367b4d157..b08a7ec079 100644
--- a/frontend/rust-lib/flowy-document/tests/document/util.rs
+++ b/frontend/rust-lib/flowy-document/tests/document/util.rs
@@ -123,6 +123,7 @@ pub async fn create_and_open_empty_document() -> (DocumentTest, Arc<MutexDocumen
     .await
     .unwrap();
 
+  test.open_document(&doc_id).await.unwrap();
   let document = test.get_document(&doc_id).await.unwrap();
 
   (test, document, data.page_id)