Fixed bug

fixed bug