diff --git a/common/lib/xmodule/xmodule/tests/__init__.py b/common/lib/xmodule/xmodule/tests/__init__.py
index 43c2bbe24d..20f31315c4 100644
--- a/common/lib/xmodule/xmodule/tests/__init__.py
+++ b/common/lib/xmodule/xmodule/tests/__init__.py
@@ -18,6 +18,7 @@ import capa.calc as calc
 import xmodule
 from xmodule.x_module import ModuleSystem
 from mock import Mock
+import xml.sax.saxutils as saxutils
 
 open_ended_grading_interface = {
         'url': 'http://sandbox-grader-001.m.edx.org/peer_grading',
diff --git a/common/lib/xmodule/xmodule/tests/test_conditional.py b/common/lib/xmodule/xmodule/tests/test_conditional.py
index 16bd222b9e..c3468905ad 100644
--- a/common/lib/xmodule/xmodule/tests/test_conditional.py
+++ b/common/lib/xmodule/xmodule/tests/test_conditional.py
@@ -106,7 +106,7 @@ class ConditionalModuleTest(unittest.TestCase):
         html = module.get_html()
         print "html type: ", type(html)
         print "html: ", html
-        html_expect = "{'ajax_url': 'courses/course_id/modx/a_location', 'element_id': 'i4x-edX-cond_test-conditional-condone', 'id': 'i4x://edX/cond_test/conditional/condone'}"
+        html_expect = "<div>{'ajax_url': 'courses/course_id/modx/a_location', 'element_id': 'i4x-edX-cond_test-conditional-condone', 'id': 'i4x://edX/cond_test/conditional/condone'}</div>"
         self.assertEqual(html, html_expect)
 
         gdi =  module.get_display_items()