Update UnitOfWork.php

Only the fix from pull request #95