Unit of work correction