Entity deselected upon clicking 'Update Entity'

[Diabolickal]

Since the Updated entity is technically a new object, it makes sense. A simple fix would be to return the new entity object then set that to the selected entity.