#64 small bug fix