diff --git a/examples/rocket_example/migration/Cargo.toml b/examples/rocket_example/migration/Cargo.toml
index 9e28446dc4..3c981c976c 100644
--- a/examples/rocket_example/migration/Cargo.toml
+++ b/examples/rocket_example/migration/Cargo.toml
@@ -9,5 +9,5 @@ name = "migration"
 path = "src/lib.rs"
 
 [dependencies]
-sea-schema = { git = "https://github.com/SeaQL/sea-schema.git", branch = "schema-manager", default-features = false, features = [ "migration", "debug-print" ] }
+sea-schema = { git = "https://github.com/SeaQL/sea-schema.git", branch = "restructure-sea-query-dep", default-features = false, features = [ "migration", "debug-print" ] }
 entity = { path = "../entity" }
diff --git a/examples/rocket_example/migration/src/m20220120_000001_create_post_table.rs b/examples/rocket_example/migration/src/m20220120_000001_create_post_table.rs
index 084e536058..f22dcdd611 100644
--- a/examples/rocket_example/migration/src/m20220120_000001_create_post_table.rs
+++ b/examples/rocket_example/migration/src/m20220120_000001_create_post_table.rs
@@ -1,5 +1,8 @@
 use entity::post::*;
-use sea_schema::{migration::*, sea_query::*};
+use sea_schema::migration::{
+    sea_query::{self, *},
+    *,
+};
 
 pub struct Migration;
 
diff --git a/sea-orm-cli/Cargo.toml b/sea-orm-cli/Cargo.toml
index bceb5c46bb..2e2a61585d 100644
--- a/sea-orm-cli/Cargo.toml
+++ b/sea-orm-cli/Cargo.toml
@@ -22,7 +22,7 @@ clap = { version = "^2.33.3" }
 dotenv = { version = "^0.15" }
 async-std = { version = "^1.9", features = [ "attributes", "tokio1" ] }
 sea-orm-codegen = { version = "^0.5.0", path = "../sea-orm-codegen" }
-sea-schema = { git = "https://github.com/SeaQL/sea-schema.git", branch = "schema-manager", default-features = false, features = [
+sea-schema = { git = "https://github.com/SeaQL/sea-schema.git", branch = "restructure-sea-query-dep", default-features = false, features = [
     "debug-print",
     "sqlx-mysql",
     "sqlx-sqlite",
diff --git a/sea-orm-cli/template/migration/Cargo.toml b/sea-orm-cli/template/migration/Cargo.toml
index 718365b83c..c55187a241 100644
--- a/sea-orm-cli/template/migration/Cargo.toml
+++ b/sea-orm-cli/template/migration/Cargo.toml
@@ -9,4 +9,4 @@ name = "migration"
 path = "src/lib.rs"
 
 [dependencies]
-sea-schema = { git = "https://github.com/SeaQL/sea-schema.git", branch = "schema-manager", default-features = false, features = [ "migration", "debug-print" ] }
+sea-schema = { git = "https://github.com/SeaQL/sea-schema.git", branch = "restructure-sea-query-dep", default-features = false, features = [ "migration", "debug-print" ] }
diff --git a/sea-orm-cli/template/migration/src/m20220101_000001_create_table.rs b/sea-orm-cli/template/migration/src/m20220101_000001_create_table.rs
index 3cb1b38cfb..df6dcf70c8 100644
--- a/sea-orm-cli/template/migration/src/m20220101_000001_create_table.rs
+++ b/sea-orm-cli/template/migration/src/m20220101_000001_create_table.rs
@@ -1,4 +1,7 @@
-use sea_schema::{migration::*, sea_query::*};
+use sea_schema::migration::{
+    sea_query::{self, *},
+    *,
+};
 
 pub struct Migration;