fixed bug
add test