Fixed bug

Fixed #160