fix #2 add test