Grid selected item disappears after reloading data?

If you look at the CRM demo online and you edit a contact:
If you cancel editing the item you were editing is still selected.
If you make a change and save the contact, the item you were editing is no longer selected.

After reloading the data the grid no longer has a selected item. How can you change this behavior? I would expect this:
If you make a change and save the contact, the item you were editing is still selected in the grid.