Update at end of unit test