fix #5 add test