diff --git a/tests/bakery_chain/baker.rs b/tests/bakery_chain/baker.rs
index e5062c8e..b359e7ef 100644
--- a/tests/bakery_chain/baker.rs
+++ b/tests/bakery_chain/baker.rs
@@ -68,4 +68,14 @@ impl Related<super::bakery::Entity> for Entity {
     }
 }
 
+impl Related<super::cake::Entity> for Entity {
+    fn to() -> RelationDef {
+        super::cakes_bakers::Relation::Cake.def()
+    }
+
+    fn via() -> Option<RelationDef> {
+        Some(super::cakes_bakers::Relation::Baker.def().rev())
+    }
+}
+
 impl ActiveModelBehavior for ActiveModel {}
diff --git a/tests/bakery_chain/bakery.rs b/tests/bakery_chain/bakery.rs
index ff4577c8..8348a764 100644
--- a/tests/bakery_chain/bakery.rs
+++ b/tests/bakery_chain/bakery.rs
@@ -67,10 +67,4 @@ impl Related<super::baker::Entity> for Entity {
     }
 }
 
-impl Model {
-    pub fn find_bakers(&self) -> Select<super::baker::Entity> {
-        Entity::find_related().belongs_to::<Entity>(self)
-    }
-}
-
 impl ActiveModelBehavior for ActiveModel {}
diff --git a/tests/bakery_chain/cake.rs b/tests/bakery_chain/cake.rs
index 976d8acb..6ba431a4 100644
--- a/tests/bakery_chain/cake.rs
+++ b/tests/bakery_chain/cake.rs
@@ -98,10 +98,4 @@ impl Related<super::baker::Entity> for Entity {
     }
 }
 
-impl Model {
-    pub fn find_bakers(&self) -> Select<super::baker::Entity> {
-        Entity::find_related().belongs_to::<Entity>(self)
-    }
-}
-
 impl ActiveModelBehavior for ActiveModel {}
diff --git a/tests/bakery_chain/customer.rs b/tests/bakery_chain/customer.rs
index edbd9311..d8566726 100644
--- a/tests/bakery_chain/customer.rs
+++ b/tests/bakery_chain/customer.rs
@@ -68,10 +68,4 @@ impl Related<super::order::Entity> for Entity {
     }
 }
 
-impl Model {
-    pub fn find_orders(&self) -> Select<super::order::Entity> {
-        Entity::find_related().belongs_to::<Entity>(self)
-    }
-}
-
 impl ActiveModelBehavior for ActiveModel {}
diff --git a/tests/bakery_chain/lineitem.rs b/tests/bakery_chain/lineitem.rs
index 261de494..9dc7e096 100644
--- a/tests/bakery_chain/lineitem.rs
+++ b/tests/bakery_chain/lineitem.rs
@@ -71,10 +71,4 @@ impl Related<super::order::Entity> for Entity {
     }
 }
 
-impl Model {
-    pub fn find_orders(&self) -> Select<super::order::Entity> {
-        Entity::find_related().belongs_to::<Entity>(self)
-    }
-}
-
 impl ActiveModelBehavior for ActiveModel {}
diff --git a/tests/bakery_chain/order.rs b/tests/bakery_chain/order.rs
index ff655699..f18c3a3b 100644
--- a/tests/bakery_chain/order.rs
+++ b/tests/bakery_chain/order.rs
@@ -93,10 +93,4 @@ impl Related<super::lineitem::Entity> for Entity {
     }
 }
 
-impl Model {
-    pub fn find_lineitems(&self) -> Select<super::lineitem::Entity> {
-        Entity::find_related().belongs_to::<Entity>(self)
-    }
-}
-
 impl ActiveModelBehavior for ActiveModel {}