Return entity after update