#79 small bug fix