Fix UOW tests