Make update operation return the updated version of the entity