UoW fixes