#136 fixed small bug