fixed bug

fixed bug