diff --git a/sea-orm-cli/Cargo.toml b/sea-orm-cli/Cargo.toml
index 27f5e093..f0643856 100644
--- a/sea-orm-cli/Cargo.toml
+++ b/sea-orm-cli/Cargo.toml
@@ -33,7 +33,7 @@ clap = { version = "^2.33.3" }
 dotenv = { version = "^0.15" }
 async-std = { version = "^1.9", features = [ "attributes", "tokio1" ] }
 sea-orm-codegen = { version = "^0.7.0", path = "../sea-orm-codegen", optional = true }
-sea-schema = { git = "https://github.com/SeaQL/sea-schema", branch = "remove-migration", default-features = false, features = [
+sea-schema = { git = "https://github.com/SeaQL/sea-schema", branch = "master", default-features = false, features = [
     "debug-print",
     "sqlx-mysql",
     "sqlx-sqlite",
diff --git a/sea-orm-cli/src/commands.rs b/sea-orm-cli/src/commands.rs
index 6a071fc2..c9a1ef00 100644
--- a/sea-orm-cli/src/commands.rs
+++ b/sea-orm-cli/src/commands.rs
@@ -117,7 +117,7 @@ pub async fn run_generate_command(matches: &ArgMatches<'_>) -> Result<(), Box<dy
                         .collect()
                 }
                 "sqlite" => {
-                    use sea_schema::sqlite::SchemaDiscovery;
+                    use sea_schema::sqlite::discovery::SchemaDiscovery;
                     use sqlx::Sqlite;
 
                     let connection = connect::<Sqlite>(max_connections, url.as_str()).await?;