Few changes in UnitOfWork specs