small bug fix...