diff --git a/tests/common/features/schema.rs b/tests/common/features/schema.rs
index 94256c05..94277604 100644
--- a/tests/common/features/schema.rs
+++ b/tests/common/features/schema.rs
@@ -1,7 +1,7 @@
 pub use super::super::bakery_chain::*;
 
 use super::*;
-use crate::common::setup::{create_table, create_enum, create_table_without_asserts};
+use crate::common::setup::{create_enum, create_table, create_table_without_asserts};
 use sea_orm::{
     error::*, sea_query, ConnectionTrait, DatabaseConnection, DbBackend, DbConn, ExecResult,
 };
diff --git a/tests/common/setup/mod.rs b/tests/common/setup/mod.rs
index 4926f68a..fae61984 100644
--- a/tests/common/setup/mod.rs
+++ b/tests/common/setup/mod.rs
@@ -108,11 +108,10 @@ where
     }
 
     let expect_stmts: Vec<Statement> = creates.iter().map(|stmt| builder.build(stmt)).collect();
-    let create_from_entity_stmts: Vec<Statement> =
-        Schema::create_enum_from_entity(entity, db.get_database_backend())
-            .iter()
-            .map(|stmt| builder.build(stmt))
-            .collect();
+    let create_from_entity_stmts: Vec<Statement> = Schema::create_enum_from_entity(entity, builder)
+        .iter()
+        .map(|stmt| builder.build(stmt))
+        .collect();
 
     assert_eq!(expect_stmts, create_from_entity_stmts);
 
@@ -133,7 +132,7 @@ where
 {
     let builder = db.get_database_backend();
     assert_eq!(
-        builder.build(&Schema::create_table_from_entity(entity)),
+        builder.build(&Schema::create_table_from_entity(entity, builder)),
         builder.build(create)
     );