Add unit test(#190)