       columnsToReplace = @grid.getOptions().numberOfColumnsToFreeze
       callback()
       cols = @grid.getColumns()
-      customColumns = @listVisibleCustomColumns().map((column) => @customColumns[@getCustomColumnId(column.id)])
+      customColumns = @listVisibleCustomColumns().map((column) => @gradebookGrid.columns.definitions[@getCustomColumnId(column.id)])
       cols.splice 0, columnsToReplace,
         @parentColumns..., customColumns...
       @grid.setColumns(cols)

      else
        assignmentColumns.concat(@aggregateColumns)

      customColumns = @listVisibleCustomColumns().map((column) => @customColumns[@getCustomColumnId(column.id)])
      frozenColumns = @parentColumns.concat(customColumns)
      frozenColumns.concat(scrollableColumns)

    ## Grid Column Definitions


      @updateColumnHeaders()

    updateColumnsAndRenderViewOptionsMenu: =>
      @setVisibleGridColumns()
      @grid.setColumns(@getVisibleGradeGridColumns())
      @updateColumnHeaders()
      @renderViewOptionsMenu()
