From 05d502d4eb4f9553cea7a2260ec4301645135dcf Mon Sep 17 00:00:00 2001
From: erikvw <ew2789@gmail.com>
Date: Tue, 12 Dec 2023 15:05:10 -0600
Subject: [PATCH] test setup

---
 edc_adverse_event/tests/tests/test_ae_and_actions.py | 4 ++--
 runtests.py                                          | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/edc_adverse_event/tests/tests/test_ae_and_actions.py b/edc_adverse_event/tests/tests/test_ae_and_actions.py
index 690fc05..c32db48 100644
--- a/edc_adverse_event/tests/tests/test_ae_and_actions.py
+++ b/edc_adverse_event/tests/tests/test_ae_and_actions.py
@@ -9,8 +9,8 @@
 from edc_list_data.site_list_data import site_list_data
 from edc_ltfu.constants import LOST_TO_FOLLOWUP
 from edc_registration.models import RegisteredSubject
+from edc_registration.utils import RegisteredSubjectDoesNotExist
 from edc_reportable import GRADE5
-from edc_sites.valid_site_for_subject_or_raise import InvalidSubjectError
 from edc_utils import get_utcnow
 from edc_visit_schedule.utils import OnScheduleError
 from model_bakery import baker
@@ -53,7 +53,7 @@ def test_subject_identifier(self):
         )
 
         self.assertRaises(
-            InvalidSubjectError,
+            RegisteredSubjectDoesNotExist,
             baker.make_recipe,
             "adverse_event_app.aeinitial",
             subject_identifier="blahblah",
diff --git a/runtests.py b/runtests.py
index 7dfa3ac..3fce08d 100644
--- a/runtests.py
+++ b/runtests.py
@@ -35,6 +35,7 @@
     EDC_NAVBAR_VERIFY_ON_LOAD=IGNORE,
     EDC_AUTH_SKIP_SITE_AUTHS=True,
     EDC_AUTH_SKIP_AUTH_UPDATER=True,
+    EDC_SITES_REGISTER_DEFAULT=True,
     INSTALLED_APPS=[
         "django.contrib.admin",
         "django.contrib.auth",