fix unit of work