Update unit test.