updated unit of work