diff --git a/mavis/test/pages/sessions/sessions_patient_page.py b/mavis/test/pages/sessions/sessions_patient_page.py
index 765bd6c7fa2..c62091b26a2 100644
--- a/mavis/test/pages/sessions/sessions_patient_page.py
+++ b/mavis/test/pages/sessions/sessions_patient_page.py
@@ -282,6 +282,7 @@ def set_up_vaccination(
         self.select_ready_for_vaccination(vaccination_record.consent_option)
         if vaccination_record.consent_option is ConsentOption.INJECTION:
             self.select_delivery_site(vaccination_record.delivery_site)
+
         self.click_continue_button()
 
         if len(notes) > MAVIS_NOTE_LENGTH_LIMIT:
diff --git a/mavis/test/pages/sessions/sessions_vaccination_wizard_page.py b/mavis/test/pages/sessions/sessions_vaccination_wizard_page.py
index 8d4e55f36b5..0fd0c192364 100644
--- a/mavis/test/pages/sessions/sessions_vaccination_wizard_page.py
+++ b/mavis/test/pages/sessions/sessions_vaccination_wizard_page.py
@@ -28,6 +28,9 @@ def __init__(self, page: Page) -> None:
             page.locator("div").filter(has_text="There is a problemEnter").nth(3)
         )
         self.continue_button = self.page.get_by_role("button", name="Continue")
+        self.select_batch_heading = self.page.get_by_role(
+            "heading", name="Which batch did you use?"
+        )
 
     @step("Click on Confirm")
     def click_confirm_button(self) -> None:
@@ -52,6 +55,10 @@ def choose_batch(self, batch_name: str) -> None:
 
         self.page.wait_for_load_state()
 
+        expect(
+            self.page.get_by_role("heading", name="Which batch did you use?")
+        ).to_be_visible()
+
         reload_until_element_is_visible(
             self.page,
             batch_radio,