diff --git a/backend/tests/document/edit.rs b/backend/tests/document/edit.rs
index 1656cc388d..cbd6be93f6 100644
--- a/backend/tests/document/edit.rs
+++ b/backend/tests/document/edit.rs
@@ -1,5 +1,5 @@
 use crate::document::helper::{DocScript, DocumentTest};
-use flowy_document::services::doc::{Document, FlowyDoc};
+use flowy_document_infra::core::{Document, FlowyDoc};
 use flowy_ot::core::{Attribute, Interval};
 
 #[rustfmt::skip]