Fix focus issue in DataGrid