diff --git a/cypress/integration/plugins/dashboards-flow-framework/create_workflow_spec.js b/cypress/integration/plugins/dashboards-flow-framework/create_workflow_spec.js
index 6eef9aee8..27711c512 100644
--- a/cypress/integration/plugins/dashboards-flow-framework/create_workflow_spec.js
+++ b/cypress/integration/plugins/dashboards-flow-framework/create_workflow_spec.js
@@ -119,7 +119,7 @@ describe('Creating Workflows Using Various Methods', () => {
       .click();
     const filePath = `cypress/fixtures/${FF_FIXTURE_BASE_PATH}semantic_search/source_data.json`;
     cy.get('input[type=file]').selectFile(filePath);
-    cy.getElementByDataTestId('closeSourceDataButton')
+    cy.getElementByDataTestId('updateSourceDataButton')
       .should('be.visible')
       .click();
     cy.mockIngestion(() => {
@@ -170,7 +170,7 @@ describe('Creating Workflows Using Various Methods', () => {
           .trigger('blur', { force: true });
       });
     });
-    cy.getElementByDataTestId('searchQueryCloseButton')
+    cy.getElementByDataTestId('updateSearchQueryButton')
       .should('be.visible')
       .click();
     cy.mockSemanticSearchIndexSearch(() => {