made a small bug fix