Correcting a small bug