diff --git a/e2e/tests/ui-driven/src/create-flow/create-flow.spec.ts b/e2e/tests/ui-driven/src/create-flow/create-flow.spec.ts
index b098fc520e..8c10cffec2 100644
--- a/e2e/tests/ui-driven/src/create-flow/create-flow.spec.ts
+++ b/e2e/tests/ui-driven/src/create-flow/create-flow.spec.ts
@@ -196,7 +196,7 @@ test.describe("Navigation", () => {
     await answerQuestion({ page, title: "Is this a test?", answer: "Yes" });
     await clickContinue({ page });
     await expect(
-      page.locator("h3", { hasText: "Yes! this is a test" }),
+      page.locator("h1", { hasText: "Yes! this is a test" }),
     ).toBeVisible();
 
     await page.getByTestId("backButton").click();
@@ -204,7 +204,7 @@ test.describe("Navigation", () => {
     await answerQuestion({ page, title: "Is this a test?", answer: "No" });
     await clickContinue({ page });
     await expect(
-      page.locator("h3", { hasText: "Sorry, this is a test" }),
+      page.locator("h1", { hasText: "Sorry, this is a test" }),
     ).toBeVisible();
   });
 });