fixed bug

fixed a bug