diff --git a/openmetadata-ui/src/main/resources/ui/cypress/e2e/Features/CustomMetric.spec.js b/openmetadata-ui/src/main/resources/ui/cypress/e2e/Features/CustomMetric.spec.js
index 3486bf46f97..bbd5b5209c9 100644
--- a/openmetadata-ui/src/main/resources/ui/cypress/e2e/Features/CustomMetric.spec.js
+++ b/openmetadata-ui/src/main/resources/ui/cypress/e2e/Features/CustomMetric.spec.js
@@ -93,7 +93,7 @@ const createCustomMetric = ({
       .click();
   }
   cy.get('[data-testid="profiler-add-table-test-btn"]').click();
-  cy.get('[data-testid="metric"]').click();
+  cy.get('[data-testid="custom-metric"]').click();
 
   // validate redirection and cancel button
   cy.get('[data-testid="heading"]').should('be.visible');
@@ -113,7 +113,7 @@ const createCustomMetric = ({
 
   // Click on create custom metric button
   cy.get('[data-testid="profiler-add-table-test-btn"]').click();
-  cy.get('[data-testid="metric"]').click();
+  cy.get('[data-testid="custom-metric"]').click();
   cy.get('[data-testid="submit-button"]').click();
 
   validateForm(isColumnMetric);
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/TableProfiler/Component/ColumnProfileTable.tsx b/openmetadata-ui/src/main/resources/ui/src/components/TableProfiler/Component/ColumnProfileTable.tsx
index 56cdee6032c..2ae58fa9892 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/TableProfiler/Component/ColumnProfileTable.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/TableProfiler/Component/ColumnProfileTable.tsx
@@ -268,7 +268,7 @@ const ColumnProfileTable = () => {
   const addButtonContent = [
     {
       label: <TabsLabel id="test-case" name={t('label.test-case')} />,
-      key: '1',
+      key: 'test-case',
       onClick: () => {
         history.push({
           pathname: getAddDataQualityTableTestPath(
@@ -280,8 +280,8 @@ const ColumnProfileTable = () => {
       },
     },
     {
-      label: <TabsLabel id="metric" name={t('label.custom-metric')} />,
-      key: '2',
+      label: <TabsLabel id="custom-metric" name={t('label.custom-metric')} />,
+      key: 'custom-metric',
       onClick: () => {
         history.push({
           pathname: getAddCustomMetricPath(