diff --git a/cms/djangoapps/contentstore/management/commands/tests/test_export_olx.py b/cms/djangoapps/contentstore/management/commands/tests/test_export_olx.py index a508505fa3..021f9fcb9a 100644 --- a/cms/djangoapps/contentstore/management/commands/tests/test_export_olx.py +++ b/cms/djangoapps/contentstore/management/commands/tests/test_export_olx.py @@ -29,7 +29,10 @@ class TestArgParsingCourseExportOlx(unittest.TestCase): """ Test export command with no arguments """ - errstring = "Error: too few arguments" + if six.PY2: + errstring = "Error: too few arguments" + else: + errstring = "Error: the following arguments are required: course_id" with self.assertRaisesRegexp(CommandError, errstring): call_command('export_olx')