add test
bug fixed