Increased code coverage for unit of work component