correct a small bug.