Add unit test.