Corrected a small bug