#130 fixed small bug