Add more unit test.