diff --git a/cognee/tasks/cleanup/cleanup_unused_data.py b/cognee/tasks/cleanup/cleanup_unused_data.py
index 175452a0a..a90d96b5c 100644
--- a/cognee/tasks/cleanup/cleanup_unused_data.py
+++ b/cognee/tasks/cleanup/cleanup_unused_data.py
@@ -18,6 +18,7 @@ from cognee.modules.data.models import Data, DatasetData
 from cognee.shared.logging_utils import get_logger    
 from sqlalchemy import select, or_  
 import cognee  
+import sqlalchemy as sa
     
 logger = get_logger(__name__)