Generic Unit Of Work enchancement