diff --git a/app/models/ai_generation_run.py b/app/models/ai_generation_run.py
index 1e0c608..7bcec15 100644
--- a/app/models/ai_generation_run.py
+++ b/app/models/ai_generation_run.py
@@ -68,6 +68,7 @@ class AIGenerationRun(Base):
     generated_items: Mapped[list["Item"]] = relationship(
         "Item",
         back_populates="generation_run",
+        primaryjoin="AIGenerationRun.id == Item.generation_run_id",
         foreign_keys="Item.generation_run_id",
         lazy="selectin",
     )