diff --git a/cognee/api/v1/datasets/routers/get_datasets_router.py b/cognee/api/v1/datasets/routers/get_datasets_router.py
index ca738dfbe..22866c30f 100644
--- a/cognee/api/v1/datasets/routers/get_datasets_router.py
+++ b/cognee/api/v1/datasets/routers/get_datasets_router.py
@@ -50,6 +50,7 @@ class DataDTO(OutDTO):
     mime_type: str
     raw_data_location: str
     dataset_id: UUID
+    label: Optional[str] = None
 
 
 class GraphNodeDTO(OutDTO):
diff --git a/cognee/modules/data/models/Data.py b/cognee/modules/data/models/Data.py
index 8a3d5e218..76b600100 100644
--- a/cognee/modules/data/models/Data.py
+++ b/cognee/modules/data/models/Data.py
@@ -56,5 +56,6 @@ class Data(Base):
             "createdAt": self.created_at.isoformat(),
             "updatedAt": self.updated_at.isoformat() if self.updated_at else None,
             "nodeSet": self.node_set,
+            "label": self.label,
             # "datasets": [dataset.to_json() for dataset in self.datasets]
         }