columnDef.cellClass() executes on an old version of the row model

In gridApi.edit.on.afterCellEdit(), I update a row property used by columnDef.cellClass(). This is used for visually displaying validation. When cellClass() executes, it is operating on an old version of the row object. $scope.apply() does not seem to execute columnDef.cellClass() again. It would be wonderful if there was a way to trigger columnDef.cellClass() for a cell, row, or the entire grid.

Reproduction Steps: Edit a cell, click away, the cell class is not applied because the row object has not been updated when columnDef.cellClass() runs.