If a currently focused column gets removed from data grid, the focus isn't captured anymore and returns to the document body.

Data grid should recognize and handle this case by moving the focus to the grid root element.