Fixed bug

Fixed bug