Update unit test stuff.