diff --git a/cms/djangoapps/contentstore/management/commands/tests/test_fix_not_found.py b/cms/djangoapps/contentstore/management/commands/tests/test_fix_not_found.py index 5a0894b6783b6788e07c96d42705eab2b1807243..4902606b704ae6ddfff516f8790f590b6f656abc 100644 --- a/cms/djangoapps/contentstore/management/commands/tests/test_fix_not_found.py +++ b/cms/djangoapps/contentstore/management/commands/tests/test_fix_not_found.py @@ -20,7 +20,12 @@ class TestFixNotFound(ModuleStoreTestCase): """ Test fix_not_found command with no arguments """ - with self.assertRaisesRegexp(CommandError, "Error: too few arguments"): + if six.PY2: + msg = "Error: too few arguments" + else: + msg = "Error: the following arguments are required: course_id" + + with self.assertRaisesRegexp(CommandError, msg): call_command('fix_not_found') def test_fix_not_found_non_split(self):