Fix grid layout bug.
Restore selection after reloadData if possible