Updated the UnitOfWork with 50% test coverage.