Correcting small bug.