From 4a7702216301081589939edbf0eda29dd3543c34 Mon Sep 17 00:00:00 2001
From: Iakov Gan <iakov@amazon.com>
Date: Tue, 29 Aug 2023 23:17:40 +0200
Subject: [PATCH] fix cleanup

---
 cid/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cid/common.py b/cid/common.py
index d74407a3..0dcf5168 100644
--- a/cid/common.py
+++ b/cid/common.py
@@ -729,7 +729,7 @@ def cleanup(self, **kwargs):
         self.qs.discover_datasets()
         used_datasets = [x for v in self.qs.dashboards.values() for x in v.datasets.values() ]
         for v in list(self.qs._datasets.values()):
-            if v.arn not in used_datasets and click.confirm(f'Delete unused dataset {v.name}?'):
+            if v.id not in used_datasets and click.confirm(f'Delete unused dataset {v.name}?'):
                 logger.info(f'Deleting dataset {v.name} ({v.arn})')
                 self.qs.delete_dataset(v.id)
                 logger.info(f'Deleted dataset {v.name} ({v.arn})')