probably a good idea instead of growing the UoW with additional methods to handle this.