Fix DI for UnitOfWork class.