diff --git a/querybook/webapp/components/QueryRunButton/QueryRunButton.tsx b/querybook/webapp/components/QueryRunButton/QueryRunButton.tsx
index 46dda5539..0ce56e8a5 100644
--- a/querybook/webapp/components/QueryRunButton/QueryRunButton.tsx
+++ b/querybook/webapp/components/QueryRunButton/QueryRunButton.tsx
@@ -105,16 +105,15 @@ export const QueryRunButton = React.forwardRef<
             />
         );
 
-        const tableSamplingDOM = (
-            // !disabled && TABLE_SAMPLING_CONFIG.enabled && hasSamplingTables ? (
-            <TableSamplingSelector
-                sampleRate={sampleRate}
-                setSampleRate={onSampleRateChange}
-                tooltipPos={runButtonTooltipPos}
-                onTableSamplingInfoClick={onTableSamplingInfoClick}
-            />
-        );
-        // ) : null;
+        const tableSamplingDOM =
+            !disabled && TABLE_SAMPLING_CONFIG.enabled && hasSamplingTables ? (
+                <TableSamplingSelector
+                    sampleRate={sampleRate}
+                    setSampleRate={onSampleRateChange}
+                    tooltipPos={runButtonTooltipPos}
+                    onTableSamplingInfoClick={onTableSamplingInfoClick}
+                />
+            ) : null;
 
         const isRowLimitEnabled =
             queryEngineById[engineId]?.feature_params.row_limit;