fix #8 unit test