diff --git a/lms/djangoapps/course_blocks/tests/test_api.py b/lms/djangoapps/course_blocks/tests/test_api.py
index a9563b2a42..b30e7b8b03 100644
--- a/lms/djangoapps/course_blocks/tests/test_api.py
+++ b/lms/djangoapps/course_blocks/tests/test_api.py
@@ -84,7 +84,7 @@ class TestGetCourseBlocks(UserPartitionTestMixin, CourseStructureTestCase):
         #                      course
         #                    /          \
         #                   /            \
-        #        A[0, 1, 2, 3]            B
+        #                 A[0]            B
         #                                 |
         #                                 |
         #                                 O