diff --git a/app/models/ai_generation_run.py b/app/models/ai_generation_run.py
index 8d5b81f..1e0c608 100644
--- a/app/models/ai_generation_run.py
+++ b/app/models/ai_generation_run.py
@@ -68,5 +68,6 @@ class AIGenerationRun(Base):
     generated_items: Mapped[list["Item"]] = relationship(
         "Item",
         back_populates="generation_run",
+        foreign_keys="Item.generation_run_id",
         lazy="selectin",
     )
diff --git a/app/models/item.py b/app/models/item.py
index f9b5b4c..87af9ee 100644
--- a/app/models/item.py
+++ b/app/models/item.py
@@ -223,6 +223,7 @@ class Item(Base):
     generation_run: Mapped[Union["AIGenerationRun", None]] = relationship(
         "AIGenerationRun",
         back_populates="generated_items",
+        foreign_keys=[generation_run_id],
         lazy="selectin",
     )