P: Selection of original after Enhancing photos

I usually select multiple photos to Enhance. But after the enhancing is done- the selection (white boxes around images) switches to the new enhanced images. It would be fantastic if it would stay with the original photo, so they could be unflagged, removed, etc. instead of having to reselect the old ones to unflag.