diff --git a/templates/model.jinja b/templates/model.jinja
index a17eb2d..3f8c541 100644
--- a/templates/model.jinja
+++ b/templates/model.jinja
@@ -1 +1,20 @@
+{% if config.prelude %}
+use {{prelude_path}}::*;
+use sea_orm::ActiveModelBehavior;
+
+#[async_trait::async_trait]
+impl ActiveModelBehavior for {{pascalCase(table_name)}}ActiveModel {}
+
+impl {{pascalCase(table_name)}}Model {}
+
+impl {{pascalCase(table_name)}}ActiveModel {}
+{% else %}
 use {{entities_path}}::{{table_name}}::{ActiveModel, Model, Entity};
+
+#[async_trait::async_trait]
+impl ActiveModelBehavior for ActiveModel {}
+
+impl Model {}
+
+impl ActiveModel {}
+{% endif %}
diff --git a/tests/src/models/user.rs b/tests/src/models/user.rs
index 49f1a64..c7b188f 100644
--- a/tests/src/models/user.rs
+++ b/tests/src/models/user.rs
@@ -1 +1,10 @@
-use super::_entities::user::{ActiveModel, Model, Entity};
\ No newline at end of file
+
+use super::prelude::*;
+use sea_orm::ActiveModelBehavior;
+
+#[async_trait::async_trait]
+impl ActiveModelBehavior for UserActiveModel {}
+
+impl UserModel {}
+
+impl UserActiveModel {}