diff --git a/sea-orm-migration/tests/migrator/m20220118_000002_create_fruit_table.rs b/sea-orm-migration/tests/migrator/m20220118_000002_create_fruit_table.rs
index 238e4af3..51c21321 100644
--- a/sea-orm-migration/tests/migrator/m20220118_000002_create_fruit_table.rs
+++ b/sea-orm-migration/tests/migrator/m20220118_000002_create_fruit_table.rs
@@ -1,6 +1,6 @@
 use super::m20220118_000001_create_cake_table::Cake;
-use sea_orm::DbBackend;
 use sea_orm_migration::prelude::*;
+use sea_orm_migration::sea_orm::DbBackend;
 
 pub struct Migration;
 
diff --git a/sea-orm-migration/tests/migrator/m20220118_000003_seed_cake_table.rs b/sea-orm-migration/tests/migrator/m20220118_000003_seed_cake_table.rs
index 489f76ef..3638fd2c 100644
--- a/sea-orm-migration/tests/migrator/m20220118_000003_seed_cake_table.rs
+++ b/sea-orm-migration/tests/migrator/m20220118_000003_seed_cake_table.rs
@@ -1,5 +1,5 @@
-use sea_orm::{entity::prelude::*, Set};
 use sea_orm_migration::prelude::*;
+use sea_orm_migration::sea_orm::{entity::*, query::*};
 
 pub struct Migration;
 
@@ -37,7 +37,7 @@ impl MigrationTrait for Migration {
 }
 
 mod cake {
-    use sea_orm::entity::prelude::*;
+    use sea_orm_migration::sea_orm::entity::prelude::*;
 
     #[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
     #[sea_orm(table_name = "cake")]