small bug repair.