diff --git a/src/databricks/labs/ucx/assessment/dashboards.py b/src/databricks/labs/ucx/assessment/dashboards.py
index f7fbaf103b..700a93bfbf 100644
--- a/src/databricks/labs/ucx/assessment/dashboards.py
+++ b/src/databricks/labs/ucx/assessment/dashboards.py
@@ -244,7 +244,7 @@ def _list_all_legacy_queries(self) -> Iterator[LegacyQuery]:
             yield from self._get_legacy_queries(*self._include_query_ids)
         else:
             try:
-                yield from self._ws.queries_legacy.list()  # TODO: Update this to non-legacy query
+                yield from self._ws.queries_legacy.list()
             except DatabricksError as e:
                 logger.warning("Cannot list Redash queries", exc_info=e)
 
@@ -266,7 +266,7 @@ def _get_legacy_queries(self, *query_ids: str) -> Iterator[LegacyQuery]:
     def _get_legacy_query(self, query_id: str) -> LegacyQuery | None:
         """Get a legacy query."""
         try:
-            return self._ws.queries_legacy.get(query_id)  # TODO: Update this to non-legacy query
+            return self._ws.queries_legacy.get(query_id)
         except DatabricksError as e:
             logger.warning(f"Cannot get Redash query: {query_id}", exc_info=e)
             return None