Fixed API call to restore selection