Restores selection after update