small bug fix.