diff --git a/examples/tokio/Cargo.toml b/examples/tokio/Cargo.toml
index b1124aad..3178ddc4 100644
--- a/examples/tokio/Cargo.toml
+++ b/examples/tokio/Cargo.toml
@@ -8,7 +8,7 @@ edition = "2018"
 publish = false
 
 [dependencies]
-sea-orm = { path = "../../", features = [ "sqlx-all", "runtime-tokio-native-tls" ] }
+sea-orm = { path = "../../", features = [ "sqlx-all", "runtime-tokio-native-tls", "debug-print" ] }
 tokio = { version = "1", features = ["full"] }
 env_logger = { version = "^0.9" }
 log = { version = "^0.4" }
diff --git a/examples/tokio/src/cake.rs b/examples/tokio/src/cake.rs
index 21b83331..cd23fb08 100644
--- a/examples/tokio/src/cake.rs
+++ b/examples/tokio/src/cake.rs
@@ -1,57 +1,14 @@
 use sea_orm::entity::prelude::*;
 
-#[derive(Copy, Clone, Default, Debug, DeriveEntity)]
-pub struct Entity;
-
-impl EntityName for Entity {
-    fn table_name(&self) -> &str {
-        "cake"
-    }
-}
-
-#[derive(Clone, Debug, PartialEq, DeriveModel, DeriveActiveModel)]
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
+#[sea_orm(table_name = "cake")]
 pub struct Model {
+    #[sea_orm(primary_key)]
     pub id: i32,
     pub name: String,
 }
 
-#[derive(Copy, Clone, Debug, EnumIter, DeriveColumn)]
-pub enum Column {
-    Id,
-    Name,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DerivePrimaryKey)]
-pub enum PrimaryKey {
-    Id,
-}
-
-impl PrimaryKeyTrait for PrimaryKey {
-    type ValueType = i32;
-
-    fn auto_increment() -> bool {
-        true
-    }
-}
-
-#[derive(Copy, Clone, Debug, EnumIter)]
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
 pub enum Relation {}
 
-impl ColumnTrait for Column {
-    type EntityName = Entity;
-
-    fn def(&self) -> ColumnDef {
-        match self {
-            Self::Id => ColumnType::Integer.def(),
-            Self::Name => ColumnType::String(None).def(),
-        }
-    }
-}
-
-impl RelationTrait for Relation {
-    fn def(&self) -> RelationDef {
-        unreachable!()
-    }
-}
-
 impl ActiveModelBehavior for ActiveModel {}