fixed bug
small