From d7b60ecf34d7af43ae8f2c94fa5fca5c7c2e5326 Mon Sep 17 00:00:00 2001
From: "JUST.in DO IT" <justin.park@airbnb.com>
Date: Thu, 14 May 2026 14:43:07 -0700
Subject: [PATCH] fix(sqllab): missing estimate action button (#40101)

(cherry picked from commit 2b71d964cc32c7a99ef30398ccdef2d4a91df86e)
---
 .../src/SqlLab/components/EstimateQueryCostButton/index.tsx     | 2 +-
 superset/sqllab/utils.py                                        | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/superset-frontend/src/SqlLab/components/EstimateQueryCostButton/index.tsx b/superset-frontend/src/SqlLab/components/EstimateQueryCostButton/index.tsx
index b89290741ae..f64de4b8570 100644
--- a/superset-frontend/src/SqlLab/components/EstimateQueryCostButton/index.tsx
+++ b/superset-frontend/src/SqlLab/components/EstimateQueryCostButton/index.tsx
@@ -113,7 +113,7 @@ const EstimateQueryCostButton = ({
         modalBody={renderModalBody()}
         triggerNode={
           <Button
-            color="primary"
+            color="default"
             variant="text"
             style={{ height: 32, padding: '4px 15px' }}
             onClick={onClickHandler}
diff --git a/superset/sqllab/utils.py b/superset/sqllab/utils.py
index b2bdf7f8bfd..cfcf02b2c2c 100644
--- a/superset/sqllab/utils.py
+++ b/superset/sqllab/utils.py
@@ -31,6 +31,7 @@ DATABASE_KEYS = [
     "allow_cvas",
     "allow_dml",
     "allow_run_async",
+    "allows_cost_estimate",
     "allows_subquery",
     "backend",
     "database_name",