small bug fix;