diff --git a/src/npc_generation.py b/src/npc_generation.py
index 119134e..33063c0 100644
--- a/src/npc_generation.py
+++ b/src/npc_generation.py
@@ -88,10 +88,6 @@ class NPCPsychology:
     fears: list[str]
     interests: list[str]
 
-@dataclass()
-class NPCRelations:
-    """The relations for non-player characters (NPCs)."""
-    list_of_relations: list[str]
 
 @dataclass()
 class NPCRelation:
@@ -100,6 +96,12 @@ class NPCRelation:
     attitude: float
     still_exist: bool
 
+
+@dataclass()
+class NPCRelations:
+    """The relations for non-player characters (NPCs)."""
+    list_of_relations: list[NPCRelation]
+
 @dataclass()
 class NPC:
     """Class for non-player characters (NPCs)."""