diff --git a/src/databricks/labs/ucx/contexts/application.py b/src/databricks/labs/ucx/contexts/application.py
index 355366cb2b..f6ae01aac3 100644
--- a/src/databricks/labs/ucx/contexts/application.py
+++ b/src/databricks/labs/ucx/contexts/application.py
@@ -12,7 +12,7 @@
 from databricks.labs.blueprint.wheels import ProductInfo, WheelsV2
 from databricks.labs.lsql.backends import SqlBackend
 from databricks.sdk import AccountClient, WorkspaceClient, core
-from databricks.sdk.errors import NotFound
+from databricks.sdk.errors import DatabricksError
 from databricks.sdk.service import sql
 
 from databricks.labs.ucx.assessment.dashboards import DashboardOwnership
@@ -589,8 +589,8 @@ def dependency_resolver(self) -> DependencyResolver:
     def table_migration_index(self) -> TableMigrationIndex:
         try:
             index = self.tables_migrator.index()
-        except NotFound as e:
-            logger.warning("Table migration index is not found. Initializing empty index.", exc_info=e)
+        except DatabricksError as e:
+            logger.warning("Table migration cannot be loaded. Initializing empty index.", exc_info=e)
             index = TableMigrationIndex([])
         return index