Fixed error in UoW class because of failing test