Correction of a small bug