Correct small bug.