A small bug fix.