           gotAllStudents.resolve()
 
       gotCustomColumns = @getCustomColumns()
-      @gotAllData = $.when(gotCustomColumns, gotAllStudents)
+      @gotAllData = $.when(gotCustomColumns, gotAllStudents, @gotAllAssignmentGroupsPromise)
 
       @allSubmissionsLoaded.done =>
         for c in @customColumns

          @assignments[assignment.id] = assignment
      @postGradesStore.setGradeBookAssignments @assignments

    gotCourse: (course) =>
      @course = course


      @getSubmissionsChunks()
      @initHeader()

    gotAllAssignmentGroups: () =>
      assignmentGroups = @allAssignmentGroups;
      @assignmentGroups = {}
      @assignments      = {}

