fix test
fix #1 add test