diff --git a/tests/visitor-completes-quiz.spec.ts b/tests/visitor-completes-quiz.spec.ts
index ecf5747..49302ad 100644
--- a/tests/visitor-completes-quiz.spec.ts
+++ b/tests/visitor-completes-quiz.spec.ts
@@ -26,12 +26,23 @@ test("visitor accepted for treatment", async ({ page }) => {
       name: /which image best matches your hair loss/i,
     }),
   ).toBeVisible();
-
   await page
     .getByRole("button", { name: /temples/i })
     .first()
     .click();
+
+  await expect(
+    page.getByRole("heading", {
+      name: /which image best matches your hair loss/i,
+    }),
+  ).toBeVisible();
   await page.getByRole("button", { name: /no/i }).click();
+
+  await expect(
+    page.getByRole("heading", {
+      name: /breast cancer/i,
+    }),
+  ).toBeVisible();
   await page.getByRole("button", { name: /no/i }).click();
 
   await expect(