Retrieve the correct object after save/update