Fixed all the UnitOfWork things
Merge pull request #8 from jonathon-walker/FixStuff

Fixed all the UnitOfWork things