Update UnitOfWork.php