Fixed the test specifically for UOW