diff --git a/enterprise/models.py b/enterprise/models.py
index f2e0615f36..07d90cf7ec 100644
--- a/enterprise/models.py
+++ b/enterprise/models.py
@@ -78,6 +78,7 @@
     from common.djangoapps.student.models import CourseEnrollment, CourseEnrollmentAllowed
 except ImportError:
     CourseEnrollment = None
+    CourseEnrollmentAllowed = None
 
 try:
     from openedx.core.djangoapps.content.course_overviews.models import CourseOverview
diff --git a/tests/test_enterprise/api/test_views.py b/tests/test_enterprise/api/test_views.py
index 2be0f0b281..4fdb9f4b4d 100644
--- a/tests/test_enterprise/api/test_views.py
+++ b/tests/test_enterprise/api/test_views.py
@@ -67,7 +67,8 @@
 )
 from test_utils.decorators import mock_api_response
 from test_utils.factories import FAKER, PendingEnterpriseCustomerUserFactory
-from test_utils.fake_enrollment_api import get_course_details, get_default_branding_object
+from test_utils.fake_enrollment_api import get_course_details
+from test_utils.fake_enterprise_api import get_default_branding_object
 
 fake = Faker()
 
@@ -3612,12 +3613,12 @@ def test_bulk_enrollment_in_bulk_courses_pending_licenses(
                     },
                     {
                         'email': 'abc@test.com',
-                        'course_run_key': 'course-v2:edX+DemoX+Second_Demo_Course',
+                        'course_run_key': 'course-v1:HarvardX+CoolScience+2016',
                         'license_uuid': '5a88bdcade7c4ecb838f8111b68e18ac'
                     },
                     {
                         'email': 'xyz@test.com',
-                        'course_run_key': 'course-v2:edX+DemoX+Second_Demo_Course',
+                        'course_run_key': 'course-v1:HarvardX+CoolScience+2016',
                         'license_uuid': '2c58acdade7c4ede838f7111b42e18ac'
                     },
                 ]