add test code
fix can't build project