diff --git a/mavis/test/pages/import_records.py b/mavis/test/pages/import_records.py
index 8f5678556f5..271b0ccbe3b 100644
--- a/mavis/test/pages/import_records.py
+++ b/mavis/test/pages/import_records.py
@@ -230,10 +230,8 @@ def verify_upload_output(self, file_path: Path) -> None:
     @step("Select year groups {1}")
     def select_year_groups(self, *year_groups: int) -> None:
         for year_group in year_groups:
-            if year_group == 0:
-                self.page.get_by_label("Reception").check()
-            else:
-                self.page.get_by_label(text=f"Year {year_group}", exact=True).check()
+            self.page.locator(f'input[type="checkbox"][value="{year_group}"]').check()
+
         self.click_continue()
 
     def import_class_list(