ok fixed bug now need add test
test added