fix #6 add test