diff --git a/x-pack/legacy/plugins/ml/public/jobs/new_job_new/pages/components/summary_step/components/post_save_options/post_save_options.tsx b/x-pack/legacy/plugins/ml/public/jobs/new_job_new/pages/components/summary_step/components/post_save_options/post_save_options.tsx
index 5a2c5fd42bc64..a6ef18d4931b9 100644
--- a/x-pack/legacy/plugins/ml/public/jobs/new_job_new/pages/components/summary_step/components/post_save_options/post_save_options.tsx
+++ b/x-pack/legacy/plugins/ml/public/jobs/new_job_new/pages/components/summary_step/components/post_save_options/post_save_options.tsx
@@ -79,7 +79,7 @@ export const PostSaveOptions: FC<Props> = ({ jobRunner }) => {
             datafeedState === DATAFEED_STATE.STARTING || datafeedState === DATAFEED_STATE.STARTED
           }
           onClick={startJobInRealTime}
-          data-test-subj="mlButtonUseFullData3"
+          data-test-subj="mlJobWizardButtonRunInRealTime"
         >
           <FormattedMessage
             id="xpack.ml.newJob.wizard.summaryStep.postSaveOptions.startJobInRealTime"
@@ -95,7 +95,7 @@ export const PostSaveOptions: FC<Props> = ({ jobRunner }) => {
             watchCreated === true
           }
           onClick={() => setWatchFlyoutVisible(true)}
-          data-test-subj="mlButtonUseFullData"
+          data-test-subj="mlJobWizardButtonCreateWatch"
         >
           <FormattedMessage
             id="xpack.ml.newJob.wizard.summaryStep.postSaveOptions.createWatch"
diff --git a/x-pack/test/functional/services/machine_learning/job_wizard_common.ts b/x-pack/test/functional/services/machine_learning/job_wizard_common.ts
index 19a270cdab5f5..dafdd30e3a7af 100644
--- a/x-pack/test/functional/services/machine_learning/job_wizard_common.ts
+++ b/x-pack/test/functional/services/machine_learning/job_wizard_common.ts
@@ -228,7 +228,7 @@ export function MachineLearningJobWizardCommonProvider({ getService }: FtrProvid
       await retry.waitForWithTimeout(
         'job processing to finish',
         5 * 60 * 1000,
-        async () => (await testSubjects.exists('mlJobWizardButtonCreateJob')) === false
+        async () => await testSubjects.exists('mlJobWizardButtonRunInRealTime')
       );
     },
   };