diff --git a/cognee/modules/data/methods/load_or_create_datasets.py b/cognee/modules/data/methods/load_or_create_datasets.py
index 1d6ef3efb..2c9a6497c 100644
--- a/cognee/modules/data/methods/load_or_create_datasets.py
+++ b/cognee/modules/data/methods/load_or_create_datasets.py
@@ -2,7 +2,7 @@ from typing import List, Union
 from uuid import UUID
 
 from cognee.modules.data.models import Dataset
-from cognee.modules.data.methods import create_authorized_dataset
+from cognee.modules.data.methods.create_authorized_dataset import create_authorized_dataset
 from cognee.modules.data.exceptions import DatasetNotFoundError
 
 
diff --git a/examples/low_level/pipeline.py b/examples/low_level/pipeline.py
index 804e42ff7..085d313a7 100644
--- a/examples/low_level/pipeline.py
+++ b/examples/low_level/pipeline.py
@@ -73,7 +73,7 @@ def ingest_files(data: List[Any]):
 
                 new_company.departments.append(departments_data_points[department_name])
 
-    return companies_data_points.values()
+    return list(companies_data_points.values())
 
 
 async def main():