diff --git a/src/registrar/tests/test_admin.py b/src/registrar/tests/test_admin.py index 786e33659..b18db43bd 100644 --- a/src/registrar/tests/test_admin.py +++ b/src/registrar/tests/test_admin.py @@ -25,6 +25,7 @@ from registrar.models import ( Domain, DomainRequest, DomainInformation, + DraftDomain, User, DomainInvitation, Contact, diff --git a/src/registrar/tests/test_models_domain.py b/src/registrar/tests/test_models_domain.py index 93407db26..55647b604 100644 --- a/src/registrar/tests/test_models_domain.py +++ b/src/registrar/tests/test_models_domain.py @@ -432,6 +432,7 @@ class TestDomainCreation(MockEppLib): DomainInformation.objects.all().delete() DomainRequest.objects.all().delete() PublicContact.objects.all().delete() + Host.objects.all().delete() Domain.objects.all().delete() User.objects.all().delete() DraftDomain.objects.all().delete()