diff --git a/frontend/rust-lib/event-integration-test/tests/user/af_cloud_test/import_af_data_folder_test.rs b/frontend/rust-lib/event-integration-test/tests/user/af_cloud_test/import_af_data_folder_test.rs
index 97c5c4b386..ea33f14180 100644
--- a/frontend/rust-lib/event-integration-test/tests/user/af_cloud_test/import_af_data_folder_test.rs
+++ b/frontend/rust-lib/event-integration-test/tests/user/af_cloud_test/import_af_data_folder_test.rs
@@ -27,7 +27,7 @@ async fn import_appflowy_data_need_migration_test() {
     .await
     .unwrap();
   // after import, the structure is:
-  // workspace:
+  // workspace: General
   //   view: Getting Started
   //   view: 037_local
   //      view: Getting Started
@@ -35,11 +35,12 @@ async fn import_appflowy_data_need_migration_test() {
   //        view: Document2
   //   view: To-dos
 
-  let views = test.get_all_workspace_views().await;
+  let spaces = test.get_all_workspace_views().await;
+  let views = &spaces[0].child_views;
   assert_eq!(views.len(), 3);
-  assert_eq!(views[1].name, import_container_name);
+  assert_eq!(views[2].name, import_container_name);
 
-  let child_views = test.get_view(&views[1].id).await.child_views;
+  let child_views = test.get_view(&views[2].id).await.child_views;
   assert_eq!(child_views.len(), 1);
 
   let child_views = test.get_view(&child_views[0].id).await.child_views;
@@ -76,7 +77,7 @@ async fn import_appflowy_data_folder_into_new_view_test() {
     .unwrap();
 
   // after import, the structure is:
-  // workspace:
+  // workspace: General
   //   view: Getting Started
   //   view: To-dos
   //   view: 040_local
@@ -84,8 +85,11 @@ async fn import_appflowy_data_folder_into_new_view_test() {
   //        view: Document2
   //          view: Grid1
   //          view: Grid2
-  let first_level_views = test.get_all_workspace_views().await;
-  let views = &first_level_views[0].child_views;
+  let spaces = test.get_all_workspace_views().await;
+  println!("first_level_views: {:?}", spaces);
+  let views = &spaces[0].child_views;
+  println!("views: {:?}", views);
+
   assert_eq!(views.len(), 3);
   assert_eq!(views[2].name, import_container_name);
 
@@ -143,18 +147,19 @@ async fn import_appflowy_data_folder_into_current_workspace_test() {
     .await
     .unwrap();
   // after import, the structure is:
-  // workspace:
+  // workspace: General
   //   view: Getting Started
   //   view: To-dos
   //   view: Document1
   //      view: Document2
   //        view: Grid1
   //        view: Grid2
-  let views = test.get_all_workspace_views().await;
+  let spaces = test.get_all_workspace_views().await;
+  let views = &spaces[0].child_views;
   assert_eq!(views.len(), 3);
-  assert_eq!(views[1].name, "Document1");
+  assert_eq!(views[2].name, "Document1");
 
-  let document_1_child_views = test.get_view(&views[1].id).await.child_views;
+  let document_1_child_views = test.get_view(&views[2].id).await.child_views;
   assert_eq!(document_1_child_views.len(), 1);
   assert_eq!(document_1_child_views[0].name, "Document2");
 
@@ -184,9 +189,10 @@ async fn import_appflowy_data_folder_into_new_view_test2() {
     .await
     .unwrap();
 
-  let views = test.get_all_workspace_views().await;
-  assert_eq!(views.len(), 2);
-  assert_eq!(views[1].name, import_container_name);
+  let spaces = test.get_all_workspace_views().await;
+  let views = &spaces[0].child_views;
+  assert_eq!(views.len(), 3);
+  assert_eq!(views[2].name, import_container_name);
   assert_040_local_2_import_content(&test, &views[1].id).await;
 
   drop(cleaner);
@@ -236,7 +242,9 @@ async fn import_appflowy_data_folder_multiple_times_test() {
   //   To-dos
   //   040_local_2
 
-  let views = test.get_all_workspace_views().await;
+  let spaces = test.get_all_workspace_views().await;
+  let views = &spaces[0].child_views;
+
   assert_eq!(views.len(), 2);
   assert_eq!(views[1].name, import_container_name);
   assert_040_local_2_import_content(&test, &views[1].id).await;
@@ -250,15 +258,17 @@ async fn import_appflowy_data_folder_multiple_times_test() {
     .unwrap();
   // after import, the structure is:
   //   Getting Started
+  //   To-dos
   //   040_local_2
   //      Getting started
   //   040_local_2
   //      Getting started
-  let views = test.get_all_workspace_views().await;
+  let spaces = test.get_all_workspace_views().await;
+  let views = &spaces[0].child_views;
   assert_eq!(views.len(), 3);
   assert_eq!(views[2].name, import_container_name);
-  assert_040_local_2_import_content(&test, &views[1].id).await;
   assert_040_local_2_import_content(&test, &views[2].id).await;
+  assert_040_local_2_import_content(&test, &views[3].id).await;
   drop(cleaner);
 }