diff --git a/superset/common/query_context_factory.py b/superset/common/query_context_factory.py
index 3a8380bdb0c..a42d1d4ba73 100644
--- a/superset/common/query_context_factory.py
+++ b/superset/common/query_context_factory.py
@@ -121,7 +121,7 @@ class QueryContextFactory:  # pylint: disable=too-few-public-methods
         datasource: BaseDatasource,
     ) -> None:
         temporal_columns = {
-            column.column_name
+            column["column_name"] if isinstance(column, dict) else column.column_name
             for column in datasource.columns
             if (column["is_dttm"] if isinstance(column, dict) else column.is_dttm)
         }