fix UOW tests