fix a small bug.