add test
fix test