diff --git a/openassessment/xblock/utils/notifications.py b/openassessment/xblock/utils/notifications.py
index 8c21e5c7a7..756699f576 100644
--- a/openassessment/xblock/utils/notifications.py
+++ b/openassessment/xblock/utils/notifications.py
@@ -6,7 +6,7 @@
 from django.conf import settings
 from openedx_events.learning.signals import COURSE_NOTIFICATION_REQUESTED
 from openedx_events.learning.data import CourseNotificationData
-from xmodule.modulestore.django import modulestore
+from openassessment.runtime_imports.functions import modulestore
 
 logger = logging.getLogger(__name__)