diff --git a/openassessment/__init__.py b/openassessment/__init__.py
index 92427cf4de..e30397be10 100644
--- a/openassessment/__init__.py
+++ b/openassessment/__init__.py
@@ -1,4 +1,4 @@
 """
 Initialization Information for Open Assessment Module
 """
-__version__ = '4.3.1'
+__version__ = '4.4.0'
diff --git a/openassessment/runtime_imports/classes.py b/openassessment/runtime_imports/classes.py
index 63c0f93be8..427cef955f 100644
--- a/openassessment/runtime_imports/classes.py
+++ b/openassessment/runtime_imports/classes.py
@@ -18,8 +18,8 @@ def import_course_waffle_flag():
     Helper method that imports CourseWaffleFlag from edx-platform at runtime.
     https://github.com/edx/edx-platform/blob/master/openedx/core/djangoapps/waffle_utils/__init__.py#L345
     """
-    from openedx.core.djangoapps.waffle_utils.__future__ import FutureCourseWaffleFlag
-    return FutureCourseWaffleFlag
+    from openedx.core.djangoapps.waffle_utils import CourseWaffleFlag
+    return CourseWaffleFlag
 
 
 def import_external_id():