Corrected a small bug.